Home
last modified time | relevance | path

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

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