Home
last modified time | relevance | path

Searched defs:NoGradGuard (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DGradMode.h27 struct C10_API NoGradGuard : public AutoGradMode { struct
28 NoGradGuard() : AutoGradMode(/*enabled=*/false) {} in NoGradGuard() argument
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A Dgrad_mode.h9 using NoGradGuard = c10::NoGradGuard; variable
/aosp_15_r20/external/pytorch/torch/csrc/api/include/torch/
H A Dutils.h43 using NoGradGuard = at::NoGradGuard; variable