Searched refs:next_cnode (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | insert_depend_for_all_gather.cc | 45 auto next_cnode = next_node->cast<CNodePtr>(); in Run() local
47 AnfAlgo::GetInputNode(next_cnode, 0), current_node}; in Run()
49 new_input->set_abstract(AnfAlgo::GetInputNode(next_cnode, 0)->abstract()); in Run()
50 AnfAlgo::SetNodeInput(next_cnode, new_input, 0); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | merge_cast_to_op.cc | 159 auto next_cnode = next_node->cast<CNodePtr>(); in MergeCastToNextOp() local
160 auto next_op_name = AnfAlgo::GetCNodeName(next_cnode); in MergeCastToNextOp()
165 kernel_query->Query(next_cnode, &kernel_info_list); in MergeCastToNextOp()
170 …[&next_cnode, &dst_type_id, &cast_index](const std::shared_ptr<kernel::KernelBuildInfo> &candidate… in MergeCastToNextOp()
171 … return AlternativeKernelInfoForInput(next_cnode, dst_type_id, cast_index, candidate_kernel_info); in MergeCastToNextOp()
178 …MS_LOG(INFO) << "Found alternative kernel info for current anf kernel " << next_cnode->DebugString… in MergeCastToNextOp()
181 AnfAlgo::SetSelectKernelBuildInfo(*alternative_kernel_info, next_cnode.get()); in MergeCastToNextOp()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fisson/ |
| D | eliminate_concat_split.cc | 118 auto next_cnode = next->cast<CNodePtr>(); in ConcatSplitEliminate() local
119 MS_ASSERT(next_cnode != nullptr); in ConcatSplitEliminate()
120 inputs_node.push_back(next_cnode); in ConcatSplitEliminate()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 1174 auto next_cnode = FindCNode(param_node_pair.first, mirror_op_name, func_graph, 0); in InsertMirrorOps() local
1176 if (next_cnode.first) { in InsertMirrorOps()
1177 MS_EXCEPTION_IF_NULL(next_cnode.second); in InsertMirrorOps()
1182 manager->SetEdge(load_node, 1, next_cnode.second); in InsertMirrorOps()
1184 manager->SetEdge(node, static_cast<int>(index), next_cnode.second); in InsertMirrorOps()
1589 auto next_cnode = FindCNode(parameter, op_name, cnode->func_graph(), 0); in ApplyParallelOptOnParam() local
1590 if (next_cnode.first) { in ApplyParallelOptOnParam()
1591 manager->SetEdge(cnode, param_pair.second, next_cnode.second); in ApplyParallelOptOnParam()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 2444 auto next_cnode = exe_orders.at(i); in AdjustAtomicAddrCleanOrder() local
2445 auto next_cnode_name = AnfAlgo::GetCNodeName(next_cnode); in AdjustAtomicAddrCleanOrder()
2447 update_orders.emplace_back(next_cnode); in AdjustAtomicAddrCleanOrder()
|