Searched refs:new_transpose_node (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | transdata_split.cc | 70 CNodePtr new_transpose_node = nullptr; in DoSplit() local
80 new_transpose_node = NewTransOpNode(func_graph, new_transdata_node, kernel_select_, false, in DoSplit()
82 RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, new_transpose_node); in DoSplit()
83 new_replace_node = new_transpose_node; in DoSplit()
86 …new_transpose_node = NewTransOpNode(func_graph, AnfAlgo::GetInputNode(node->cast<CNodePtr>(), 0), … in DoSplit()
89 AnfAlgo::SetNodeAttr("nop_op", MakeValue(true), new_transpose_node); in DoSplit()
91 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transpose_node); in DoSplit()
95 …NewTransOpNode(func_graph, new_transpose_node, kernel_select_, false, prim::kPrimTransData->name()… in DoSplit()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_transpose_for_dynamic_gru_v2.cc | 51 AnfNodePtr new_transpose_node = nullptr; in Insert() local
58 new_transpose_node = in Insert()
61 MS_EXCEPTION_IF_NULL(new_transpose_node); in Insert()
62 AnfAlgo::SetNodeAttr("nop_op", MakeValue(true), new_transpose_node); in Insert()
63 RefreshKernelBuildInfo(input_format, kOpFormat_HWCN, new_transpose_node); in Insert()
66 …NewTransOpNode(func_graph, new_transpose_node, kernel_select, false, prim::kPrimTransData->name()); in Insert()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | lstm_adjust_pass.cc | 284 auto new_transpose_node = func_graph->NewCNode(transpose_prim, {cnode, transpose_perm}); in Run() local
285 MS_CHECK_TRUE_MSG(new_transpose_node != nullptr, false, "New transpose node failed."); in Run()
290 auto new_reshape_node = func_graph->NewCNode(reshape_prim, {new_transpose_node, reshape_perm}); in Run()
|