m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!1431 fix bug of transdata and transpose split

Browse Source 
Merge pull request !1431 from lianliguang/master
pull/1431/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
parent e5c7ecfd46 9d364f509f
commit 28029388c1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.cc
Unescape View File

@ -90,7 +90,7 @@ bool TransDataSplit::DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &n
new_transdata_node = new_transdata_node =
NewTransOpNode(func_graph, new_transpose_node, kernel_select_, false, prim::KPrimTransData->name()); NewTransOpNode(func_graph, new_transpose_node, kernel_select_, false, prim::KPrimTransData->name());
RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, AnfAlgo::GetOutputDeviceDataType(new_transdata_node, 0), RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, AnfAlgo::GetOutputDeviceDataType(new_transdata_node, 0),
new_transpose_node); new_transdata_node);
new_replace_node = new_transdata_node; new_replace_node = new_transdata_node;
} }
FuncGraphManagerPtr manager = func_graph->manager(); FuncGraphManagerPtr manager = func_graph->manager();

Write Preview
Loading…
Cancel
Save