Searched refs:transdata_node (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | insert_transpose_for_dynamic_gru_v2.cc | 52 AnfNodePtr transdata_node = AnfAlgo::GetInputNode(cnode, index); in Insert() local
53 auto input_format = AnfAlgo::GetInputFormat(transdata_node, 0); in Insert()
54 auto output_format = AnfAlgo::GetOutputFormat(transdata_node, 0); in Insert()
55 auto padding_axis = AnfAlgo::GetOutputReshapeType(transdata_node, 0); in Insert()
59 …NewTransOpNode(func_graph, AnfAlgo::GetInputNode(transdata_node->cast<CNodePtr>(), 0), kernel_sele… in Insert()
69 new_transdata_node->set_abstract(transdata_node->abstract()); in Insert()
75 if (!manager->Replace(transdata_node, new_node)) { in Insert()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_launch_transdata.cc | 99 auto transdata_node = transdata_graph_->execution_order()[0]; in ConstructKernelGraphAndSetAttr() local
101 AnfAlgo::SetOutputInferTypeAndShape({dtype_}, {shape_}, transdata_node.get()); in ConstructKernelGraphAndSetAttr()
112 AnfAlgo::SetSelectKernelBuildInfo(builder->Build(), transdata_node.get()); in ConstructKernelGraphAndSetAttr()
114 AnfAlgo::SetNodeAttr(kAttrSrcFormat, MakeValue(src_format_), transdata_node); in ConstructKernelGraphAndSetAttr()
115 AnfAlgo::SetNodeAttr(kAttrDstFormat, MakeValue(dst_format_), transdata_node); in ConstructKernelGraphAndSetAttr()
|