diff --git a/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc b/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
index 7f11c8f2c7e..58c030e79dc 100644
--- a/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
+++ b/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
@@ -94,7 +94,7 @@ AnfNodePtr GetTransInputNodePtr(const FuncGraphPtr &func_graph, const CNodePtr &
   MS_EXCEPTION_IF_NULL(node);
   bool padding_flag = false;
   auto input_node = AnfAlgo::GetInputNode(node, index);
-  if (!AnfAlgo::IsFeatureMapInput(node, index)) {
+  if (input_node->isa<ValueNode>() || input_node->isa<Parameter>()) {
     input_node = InsertTransOpForOutput(func_graph, input_node, kernel_select);
     MS_EXCEPTION_IF_NULL(input_node);
     AnfAlgo::SetNodeInput(node, input_node, index);
diff --git a/mindspore/ccsrc/session/ascend_session.cc b/mindspore/ccsrc/session/ascend_session.cc
index f255b2f15fd..751cf76e322 100755
--- a/mindspore/ccsrc/session/ascend_session.cc
+++ b/mindspore/ccsrc/session/ascend_session.cc
@@ -186,7 +186,7 @@ void AscendSession::RunGraph(const GraphId &graph_id, const std::vector<tensor::
 void AscendSession::RunOpHardwareOptimize(const std::shared_ptr<session::KernelGraph> &kernel_graph) const {
   MS_LOG(INFO) << "Start";
   // data layout optimization
-  opt::AscendDataLayout(kernel_graph);
+  opt::RunOpAscendDataLayout(kernel_graph);
   // mixed precision optimization
   opt::AscendMixPrecision(kernel_graph);
   MS_LOG(INFO) << "Finish";