Searched refs:_VirtualAdd (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ops/operations/ |
| D | __init__.py | 40 … _VirtualOutput, _VirtualDiv, _GetTensorSlice, _VirtualAdd, _VirtualAssignAdd, _VirtualAccuGrad,
|
| D | comm_ops.py | 795 class _VirtualAdd(PrimitiveWithInfer): class
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_comm_ops.py | 27 … ReduceScatter, _HostReduceScatter, _VirtualDiv, _VirtualAdd, AllSwap,
118 @bprop_getters.register(_VirtualAdd)
|