| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/ |
| D | grad_reducer.py | 117 def _tensors_allreduce(degree, mean, allgather, allreduce, allreduce_filter, grad): argument
162 def _tensors_allreduce_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, ps_parameter): argument
190 def _tensors_allreduce_with_sparse(degree, mean, allgather, allreduce, allreduce_filter, grad): argument
216 def _tensors_allreduce_with_sparse_ps(degree, mean, allgather, allreduce, allreduce_filter, grad, p… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/boost/ |
| D | adasum.py | 55 def _send_recv_res(left_send, recv_part, local_part, allreduce, parameter_divisibility, allreduce_n… argument
85 def _adasum_opt_forward_process(left_send, allreduce, parameter_divisibility, allreduce_node_num, s… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ |
| D | adasum.py | 75 def _send_recv_res(left_send, recv_part, local_part, allreduce, parameter_divisibility, allreduce_n… argument
112 def _adasum_opt_forward_process(left_send, allreduce, parameter_divisibility, allreduce_node_num, s… argument
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | ir_fusion_test.py | 23 allreduce = P.AllReduce() variable
|
| D | bn_grad_split.py | 31 allreduce = Primitive('AllReduce') variable
|
| D | bn_split.py | 32 allreduce = Primitive('AllReduce') variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | bn_split.cc | 340 auto allreduce = pass.NewCNode(allreduce_inputs, graph); in CreateAllReduceAndMul() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | matmul_allreduce_fusion.cc | 127 auto allreduce = VectorRef({is_allreduce, quant_bmm}); in DefineQuantBatchMatmulAllReducePattern() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 3530 CNodePtr allreduce = root->NewCNode({NewValueNode(prim::kPrimAllReduce), tuple_getitem}); in BroadcastMultiOutputs() local
|