Searched refs:_VirtualAccuGrad (Results 1 – 3 of 3) sorted by relevance
851 class _VirtualAccuGrad(PrimitiveWithInfer): class868 virtual_accu_grad = _VirtualAccuGrad()
40 … _VirtualOutput, _VirtualDiv, _GetTensorSlice, _VirtualAdd, _VirtualAssignAdd, _VirtualAccuGrad,
28 … _VirtualAssignAdd, _VirtualAccuGrad, _MirrorMicroStepOperator, _MicroStepAllGather)141 @bprop_getters.register(_VirtualAccuGrad)