Searched defs:comm_node (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | bias_add_comm_swap.cc | 42 bool IsAddNodeValid(const CNodePtr &add_node, const AnfNodePtr &comm_node) { in IsAddNodeValid()
79 auto comm_node = GetInputNodeWithFilter(node, [&](const AnfNodePtr &anode) { in FindValidCommNode() local
108 auto comm_node = FindValidCommNode(node); in FindAllValidAddNode() local
122 void HandleNodePullUp(const AnfNodePtr &comm_node, const CNodePtr &add_node) { in HandleNodePullUp()
151 bool HandleNodeBiasAdd(const AnfNodePtr &comm_node, const CNodePtr &add_node) { in HandleNodeBiasAdd()
215 void HandleNodePullDown(const AnfNodePtr &comm_node, const CNodePtr &add_node) { in HandleNodePullDown()
232 auto comm_node = node_pair.second; in HandleAddNode() local
|
| D | matmul_add_comm_reduction.cc | 53 bool IsAddNodeValid(const AnfNodePtr &add_node, const AnfNodePtr &comm_node) { in IsAddNodeValid()
127 auto comm_node = FindPullDownNode(input_node); in FindAllValidAddNode() local
155 AnfNodePtr FindBiasAdd(const AnfNodePtr &comm_node, const AnfNodePtr &add_node_input) { in FindBiasAdd()
171 void HandleNodeBiasAdd(const AnfNodePtr &comm_node, const AnfNodePtr &add_node_input) { in HandleNodeBiasAdd()
271 void HandleNodePullDown(const AnfNodePtr &add_node, const AnfNodePtr &comm_node) { in HandleNodePullDown()
|
| D | split_matmul_comm_elementwise_fp.cc | 87 const AnfNodePtr &comm_node) { in SplitIntoInterleaved()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/ |
| D | insert_depend_for_grad_comm.cc | 311 for (const auto &comm_node : current_node_list) { in Run() local
318 for (const auto &comm_node : current_node_list) { in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | operator_info.cc | 870 int32_t AddCommOpFusionType(const CNodePtr &comm_node, const AnfNodePtr ¶m_node) { in AddCommOpFusionType()
902 void AddCommOpMeanFlag(const CNodePtr &comm_node) { in AddCommOpMeanFlag()
912 void AddCNodePrimAttr(const CNodePtr &comm_node, const std::string &attr_name, const ValuePtr &attr… in AddCNodePrimAttr()
920 void AddCommOpParamFlag(const CNodePtr &comm_node) { in AddCommOpParamFlag()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 1765 void AddNodeFusionInfo(const CNodePtr &node, const CNodePtr &comm_node, const std::string &backward… in AddNodeFusionInfo()
|