Searched refs:_MirrorMicroStepOperator (Results 1 – 3 of 3) sorted by relevance
41 … _HostAllGather, _HostReduceScatter, _MirrorMicroStepOperator, _MicroStepAllGather)
871 class _MirrorMicroStepOperator(PrimitiveWithInfer): class
28 … _VirtualAssignAdd, _VirtualAccuGrad, _MirrorMicroStepOperator, _MicroStepAllGather)154 @bprop_getters.register(_MirrorMicroStepOperator)