Searched defs:AutoGradMode (Results 1 – 4 of 4) sorted by relevance
15 struct C10_API AutoGradMode { struct16 AutoGradMode(bool enabled) : prev_mode(GradMode::is_enabled()) { in AutoGradMode() argument19 ~AutoGradMode() { in ~AutoGradMode() argument27 struct C10_API NoGradGuard : public AutoGradMode { argument
9 using AutoGradMode = at::AutoGradMode; variable
8 using AutoGradMode = c10::AutoGradMode; variable
71 using AutoGradMode = at::AutoGradMode; variable