Searched defs:allgather (Results 1 – 2 of 2) 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/ccsrc/frontend/parallel/ |
| D | step_parallel.cc | 1530 CNodePtr allgather; in InsertAllGatherOp() local
|