version_converter/adapters/cast_9_8.h

0001 // Copyright (c) ONNX Project Contributors
0002
0003 /*
0004  * SPDX-License-Identifier: Apache-2.0
0005  */
0006
0007 // Adapter for Cast in default domain from version 9 to 8
0008
0009 #pragma once
0010
0011 #include <memory>
0012
0013 #include "onnx/version_converter/adapters/adapter.h"
0014
0015 namespace ONNX_NAMESPACE {
0016 namespace version_conversion {
0017
0018 class Cast_9_8 final : public Adapter {
0019  public:
0020   explicit Cast_9_8() : Adapter("Cast", OpSetID(9), OpSetID(8)) {}
0021
0022   void adapt_cast_9_8(std::shared_ptr<Graph>, Node* node) const {
0023     if (node->inputs()[0]->elemType() == TensorProto_DataType_STRING || node->i(kto) == TensorProto_DataType_STRING)
0024       ONNX_ASSERTM(false, "Casting From/To STRING data type is not supported")
0025   }
0026
0027   Node* adapt(std::shared_ptr<Graph> graph, Node* node) const override {
0028     adapt_cast_9_8(graph, node);
0029     return node;
0030   }
0031 };
0032
0033 } // namespace version_conversion
0034 } // namespace ONNX_NAMESPACE