Searched defs:allreduce (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | grad_reducer.py | 83 def _tensors_allreduce(degree, mean, allgather, allreduce, allreduce_filter, grad): argument
130 def _tensors_allreduce_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, ps_parameter): argument
158 def _tensors_allreduce_with_sparse(degree, mean, allgather, allreduce, allreduce_filter, grad): argument
184 def _tensors_allreduce_with_sparse_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, p… argument
|
| /third_party/mindspore/mindspore/boost/ |
| D | adasum.py | 56 def _send_recv_res(left_send, recv_part, local_part, allreduce, parameter_divisibility, allreduce_n… argument
88 def _adasum_opt_forward_process(left_send, allreduce, parameter_divisibility, allreduce_node_num, s… argument
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | bn_grad_split.py | 31 allreduce = Primitive('AllReduce') variable
|
| D | ir_fusion_test.py | 23 allreduce = P.AllReduce() variable
|
| D | bn_split.py | 32 allreduce = Primitive('AllReduce') variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.h | 346 …void InsertFromSendRecvPair(const CNodePtr &allreduce, const std::pair<CNodePtr, CNodePtr> &send_r… in InsertFromSendRecvPair()
349 …void InsertToSendRecvPair(const CNodePtr &allreduce, const std::pair<CNodePtr, CNodePtr> &send_rec… in InsertToSendRecvPair()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | bn_split.cc | 217 auto allreduce = graph->NewCNode(allreduce_inputs); in CreateAllReduceAndMul() local
|