From 8bbb3589b2551c45d5aa2d14390637929dde0928 Mon Sep 17 00:00:00 2001
From: wangxiaotian22 <wangxiaotian4@huawei.com>
Date: Thu, 19 Nov 2020 09:32:15 +0800
Subject: [PATCH] remove mark format in tensorflow parser

---
 parser/tensorflow/tensorflow_parser.cc | 28 --------------------------
 1 file changed, 28 deletions(-)

diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc
index 1800d00..7844bfd 100644
--- a/parser/tensorflow/tensorflow_parser.cc
+++ b/parser/tensorflow/tensorflow_parser.cc
@@ -181,17 +181,6 @@ const std::vector<std::string> kSkipCheckoutInputSizeNodes = {ge::parser::DATA,
                                                               ge::parser::FRAMEWORKOP, ge::parser::LAYERNORM};
 const std::vector<std::string> kMakeOperatorNotByIr = {ge::parser::ARG, ge::parser::VARIABLE, ge::parser::VARHANDLEOP,
                                                        ge::parser::FRAMEWORKOP, ge::parser::DATA};
-const std::map<string, domiTensorFormat_t> kNeedMarkFormatNodes = {
-    {"ExtractImagePatches", domi::DOMI_TENSOR_NHWC},
-    {"ExtractVolumePatches", domi::DOMI_TENSOR_NHWC},
-    {"LogSoftmax", domi::DOMI_TENSOR_NHWC},
-    {"ResizeBilinear", domi::DOMI_TENSOR_NHWC},
-    {"ResizeBilinearGrad", domi::DOMI_TENSOR_NHWC},
-    {"ResizeNearestNeighbor", domi::DOMI_TENSOR_NHWC},
-    {"Softmax", domi::DOMI_TENSOR_NHWC},
-    {"SoftmaxCrossEntropyWithLogits", domi::DOMI_TENSOR_NHWC},
-    {"SoftmaxGrad", domi::DOMI_TENSOR_NHWC},
-    {"SpaceToBatch", domi::DOMI_TENSOR_NHWC}};
 const char *const kDpop = "DPOP";
 const char *const kFuncDefLibraryFilePath = "graph_def_library.pbtxt";
 const char *const kAttrNameIsScopeInnerNode = "_is_scope_inner_node";
@@ -1028,23 +1017,6 @@ Status TensorFlowModelParser::AddNodeToGraphAndMarkFormat(ge::ComputeGraphPtr &g
     graph->AddNode(iterator->second);
   }
 
-  // mark format with default one explained in tf documents for some nodes
-  for (auto &node : graph->GetDirectNode()) {
-    auto nodeType = node->GetType();
-    auto iter = kNeedMarkFormatNodes.find(nodeType);
-    if (iter != kNeedMarkFormatNodes.end()) {
-      GE_CHECK_NOTNULL(node->GetOpDesc());
-      for (auto &input_desc : node->GetOpDesc()->GetAllInputsDescPtr()) {
-        input_desc->SetOriginFormat((ge::Format)iter->second);
-        input_desc->SetFormat((ge::Format)iter->second);
-      }
-      for (auto &output_desc : node->GetOpDesc()->GetAllOutputsDescPtr()) {
-        output_desc->SetOriginFormat((ge::Format)iter->second);
-        output_desc->SetFormat((ge::Format)iter->second);
-      }
-    }
-  }
-
   return SUCCESS;
 }