From 0f68aa55834dc61aec6bda4685c2bbca6e3f5aa3 Mon Sep 17 00:00:00 2001
From: huanruizhi <huanruizhi@hisilicon.com>
Date: Sat, 13 Aug 2022 09:57:29 +0800
Subject: [PATCH] add const automapping

---
 parser/tensorflow/tensorflow_constant_parser.cc | 4 ++++
 1 file changed, 4 insertions(+)

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<const NodeDef *>(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());