diff --git a/metadef b/metadef index 3c1c28b..1c8737f 160000 --- a/metadef +++ b/metadef @@ -1 +1 @@ -Subproject commit 3c1c28b3f0f31404933cfc1bd3cad28618b09317 +Subproject commit 1c8737f98c45811dace5ae16367b947dadc55f1d diff --git a/parser/tensorflow/tensorflow_constant_parser.cc b/parser/tensorflow/tensorflow_constant_parser.cc index 86cf888..20b612b 100644 --- a/parser/tensorflow/tensorflow_constant_parser.cc +++ b/parser/tensorflow/tensorflow_constant_parser.cc @@ -28,6 +28,7 @@ #include "parser/common/op_parser_factory.h" #include "framework/omg/parser/parser_types.h" #include "register/tensor_assign.h" +#include "register/register_utils.h" using domi::tensorflow::NodeDef; using domi::TENSORFLOW; @@ -90,6 +91,9 @@ Status TensorFlowConstantParser::ParseParams(const Message *op_src, ge::OpDescPt const NodeDef *node = DOMI_DYNAMIC_CAST(op_src); GE_CHECK_NOTNULL(node); GELOGD("TF op node name = %s, op type= %s, parse params", node->name().c_str(), node->op().c_str()); + ge::Operator org_op = ge::OpDescUtils::CreateOperatorFromOpDesc(op_dest); + GE_CHK_STATUS_RET(domi::OperatorAutoMapping(op_src, org_op), + "[Call][AutoMapping] failed."); ConstantOperator op; op.Name(node->name());