Home
last modified time | relevance | path

Searched defs:DistAccumulateGradCaptureHook (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/distributed/autograd/context/
Dcontext.h84 friend class DistAccumulateGradCaptureHook; variable
/external/pytorch/torch/csrc/distributed/autograd/engine/
Ddist_engine.cpp37 class DistAccumulateGradCaptureHook class
40 DistAccumulateGradCaptureHook( in DistAccumulateGradCaptureHook() function in torch::distributed::autograd::DistAccumulateGradCaptureHook