From 9d5890d9b99da399b618a6f78556673c3fcc26c9 Mon Sep 17 00:00:00 2001
From: lianliguang <lianliguang@huawei.com>
Date: Tue, 31 Mar 2020 09:35:44 +0800
Subject: [PATCH] fix bug of got a error transdata's dest format

---
 mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc b/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
index 6fff8ccd2a..7f11c8f2c7 100644
--- a/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
+++ b/mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
@@ -187,10 +187,10 @@ AnfNodePtr InsertTransOpForSingleOutput(const FuncGraphPtr &func_graph, const An
   return node;
 }
 
-void GetTransDataInputFormat(const AnfNodePtr &node, std::string *input_format) {
+void GetTransDataInputFormat(const AnfNodePtr &node, size_t idx, std::string *input_format) {
   MS_EXCEPTION_IF_NULL(input_format);
   if (AnfAlgo::IsRealKernel(node)) {
-    *input_format = AnfAlgo::GetOutputFormat(node, 0);
+    *input_format = AnfAlgo::GetOutputFormat(node, idx);
   } else {
     *input_format = AnfAlgo::GetPrevNodeOutputFormat(node, 0);
   }
@@ -206,7 +206,7 @@ AnfNodePtr InsertTransOpForMultipleOutput(const FuncGraphPtr &func_graph, const
     bool padding_flag = false;
 
     std::string output_format;
-    GetTransDataInputFormat(node, &output_format);
+    GetTransDataInputFormat(node, output_idx, &output_format);
     if (output_format == kOpFormat_NC1KHKWHWC0) {
       MS_LOG(EXCEPTION) << "got the hw format" << output_format << " when insert the transdata node "
                         << node->DebugString();