Searched refs:_MirrorMiniStepOperator (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 742 class _MirrorMiniStepOperator(PrimitiveWithInfer): class
769 mirror_mini_step = _MirrorMiniStepOperator()
|
| D | __init__.py | 39 … _MirrorOperator, _MirrorMiniStepOperator, _MiniStepAllGather, ReduceOp, _VirtualDataset,
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_comm_ops.py | 26 … Broadcast, _GetTensorSlice, _MirrorOperator, _MirrorMiniStepOperator, ReduceOp,
443 @bprop_getters.register(_MirrorMiniStepOperator)
|