Searched defs:accumulateGrad (Results 1 – 3 of 3) sorted by relevance
115 static void accumulateGrad( in accumulateGrad() function
71 void DistAutogradContext::accumulateGrad( in accumulateGrad() function in torch::distributed::autograd::DistAutogradContext
41 std::shared_ptr<AccumulateGrad> accumulateGrad, in DistAccumulateGradCaptureHook()