|
|
|
@@ -88,6 +88,7 @@ void STestOnnxParser::RegisterCustomOp() { |
|
|
|
REGISTER_CUSTOM_OP("Add") |
|
|
|
.FrameworkType(domi::ONNX) |
|
|
|
.OriginOpType("ai.onnx::11::Add") |
|
|
|
.OriginOpType("ai.onnx::17::Add") |
|
|
|
.ParseParamsFn(ParseParams); |
|
|
|
|
|
|
|
REGISTER_CUSTOM_OP("Identity") |
|
|
|
@@ -174,4 +175,21 @@ TEST_F(STestOnnxParser, onnx_parser_const_data_type) { |
|
|
|
EXPECT_EQ(ret, GRAPH_SUCCESS); |
|
|
|
} |
|
|
|
|
|
|
|
REG_OP(FileConstant) |
|
|
|
.OUTPUT(y, TensorType(DT_INT32)) |
|
|
|
.REQUIRED_ATTR(file_id, String) |
|
|
|
.REQUIRED_ATTR(shape, ListInt) |
|
|
|
.REQUIRED_ATTR(dtype, Type) |
|
|
|
.OP_END_FACTORY_REG(FileConstant) |
|
|
|
|
|
|
|
TEST_F(STestOnnxParser, onnx_parser_file_const_data_type) { |
|
|
|
std::string case_dir = __FILE__; |
|
|
|
case_dir = case_dir.substr(0, case_dir.find_last_of("/")); |
|
|
|
std::string model_file = case_dir + "/origin_models/onnx_file_const_type.onnx"; |
|
|
|
std::map<ge::AscendString, ge::AscendString> parser_params; |
|
|
|
ge::Graph graph; |
|
|
|
auto ret = ge::aclgrphParseONNX(model_file.c_str(), parser_params, graph); |
|
|
|
EXPECT_EQ(ret, GRAPH_SUCCESS); |
|
|
|
} |
|
|
|
|
|
|
|
} // namespace ge |