From f0cc6d4bfabbc6d80c475b82d81a324abf5b4163 Mon Sep 17 00:00:00 2001 From: PengC Date: Sat, 23 Nov 2024 14:35:31 +0800 Subject: [PATCH] AddV2 --- .../framework/tf_plugin/tensorflow_add_custom_plugin.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/operator/ascendc/0_introduction/1_add_frameworklaunch/AddCustom/framework/tf_plugin/tensorflow_add_custom_plugin.cc b/operator/ascendc/0_introduction/1_add_frameworklaunch/AddCustom/framework/tf_plugin/tensorflow_add_custom_plugin.cc index 2cd837ce5..690147bf3 100644 --- a/operator/ascendc/0_introduction/1_add_frameworklaunch/AddCustom/framework/tf_plugin/tensorflow_add_custom_plugin.cc +++ b/operator/ascendc/0_introduction/1_add_frameworklaunch/AddCustom/framework/tf_plugin/tensorflow_add_custom_plugin.cc @@ -18,6 +18,6 @@ namespace domi { // register op info to GE REGISTER_CUSTOM_OP("AddCustom") .FrameworkType(TENSORFLOW) // type: CAFFE, TENSORFLOW - .OriginOpType("Add") // name in tf module + .OriginOpType("AddV2") // name in tf module .ParseParamsByOperatorFn(AutoMappingByOpFn); } // namespace domi -- Gitee