Home
last modified time | relevance | path

Searched defs:next_cnode (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Ddropout_gen_mask_depend.cc71 auto next_cnode = next_node->cast<CNodePtr>(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_depend_for_all_gather.cc51 auto next_cnode = next_node->cast<CNodePtr>(); in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doptimize_parallel_allgather_comm.cc82 auto next_cnode = cast_next_node_user_pair.first->cast<CNodePtr>(); in MoveCastBehindAllGather() local
Dsplit_layernorm_comm_fp.cc224 auto next_cnode = next_node->cast<CNodePtr>(); in PatternFilter() local
286 auto next_cnode = node_users.front().first->cast<CNodePtr>(); in ExpandSliceRangeToRight() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/
Deliminate_concat_split.cc129 auto next_cnode = next->cast<CNodePtr>(); in ConcatSplitEliminate() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/
Dtensor_redistribution.h53 void SetPreAndNextCNode(const AnfNodePtr &pre_cnode, const CNodePtr &next_cnode) { in SetPreAndNextCNode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dmerge_cast_to_op.cc171 auto next_cnode = next_node->cast<CNodePtr>(); in MergeCastToNextOp() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/
Dstep_parallel.cc388 … const std::pair<AnfNodePtr, std::vector<int>> &node_pair, const CNodePtr &next_cnode, in ObtainOutputTensorLayout()
430 auto next_cnode = node_pair.first->cast<CNodePtr>(); in InsertRedistributionForMicroInterleaved() local
490 auto next_cnode = node_pair.first->cast<CNodePtr>(); in Redistribution() local
1245 auto next_cnode = FindCNode(param_node_pair.first, mirror_op_name, func_graph, 0); in DoInsertMirrorOps() local
1545 auto next_cnode = FindCNode(parameter, op_name, cnode->func_graph(), 0); in InsertParallelOpt() local
Dstep_parallel_utils.cc1791 auto next_cnode = pair.first->cast<CNodePtr>(); in AddNodeFusionInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dnorm_fusion.cc528 auto next_cnode = node_user.first->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dacl_stream_assign.cc215 CNodePtr pre_cnode, CNodePtr next_cnode) { in AddBoundarySendRecvKernel()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/graph_compiler/
Dgraph_partition.cc715 auto next_cnode = dyn_cast_ptr<CNode>(next_node); in ProcessCloseFollowing() local