| @@ -28,6 +28,7 @@ | |||||
| #include "parser/common/op_parser_factory.h" | #include "parser/common/op_parser_factory.h" | ||||
| #include "framework/omg/parser/parser_types.h" | #include "framework/omg/parser/parser_types.h" | ||||
| #include "register/tensor_assign.h" | #include "register/tensor_assign.h" | ||||
| #include "register/register_utils.h" | |||||
| using domi::tensorflow::NodeDef; | using domi::tensorflow::NodeDef; | ||||
| using domi::TENSORFLOW; | using domi::TENSORFLOW; | ||||
| @@ -90,6 +91,9 @@ Status TensorFlowConstantParser::ParseParams(const Message *op_src, ge::OpDescPt | |||||
| const NodeDef *node = DOMI_DYNAMIC_CAST<const NodeDef *>(op_src); | const NodeDef *node = DOMI_DYNAMIC_CAST<const NodeDef *>(op_src); | ||||
| GE_CHECK_NOTNULL(node); | GE_CHECK_NOTNULL(node); | ||||
| GELOGD("TF op node name = %s, op type= %s, parse params", node->name().c_str(), node->op().c_str()); | 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; | ConstantOperator op; | ||||
| op.Name(node->name()); | op.Name(node->name()); | ||||