Searched refs:AutoGradMode (Results 1 – 22 of 22) sorted by relevance
15 struct C10_API AutoGradMode { struct16 AutoGradMode(bool enabled) : prev_mode(GradMode::is_enabled()) { in AutoGradMode() function19 ~AutoGradMode() { in ~AutoGradMode() argument27 struct C10_API NoGradGuard : public AutoGradMode { argument28 NoGradGuard() : AutoGradMode(/*enabled=*/false) {} in NoGradGuard()
9 torch::AutoGradMode mode(false); in TEST()20 torch::AutoGradMode mode(false); in TEST()30 torch::AutoGradMode mode(false); in TEST()47 torch::AutoGradMode mode(false); in TEST()
860 at::AutoGradMode enable_grad(true); in TEST()876 at::AutoGradMode enable_grad(true); in TEST()896 at::AutoGradMode enable_grad(true); in TEST()909 at::AutoGradMode enable_grad(true); in TEST()939 at::AutoGradMode enable_grad(true); in TEST()953 at::AutoGradMode enable_grad(true); in TEST()
468 AutoGradMode mode(false); in TEST()
9 using AutoGradMode = at::AutoGradMode; variable
249 AutoGradMode grad_mode(false); in _view_as_self_with_no_grad()
354 AutoGradMode grad_mode(false);
699 AutoGradMode grad_mode(true); in grad_fn()
1058 AutoGradMode grad_mode(false); in evaluate_function()
1344 AutoGradMode grad_mode(false); in THPFunction_apply()
8 using AutoGradMode = c10::AutoGradMode; variable
71 using AutoGradMode = at::AutoGradMode; variable
83 at::AutoGradMode enable_grad(true); in step()
66 at::AutoGradMode enable_grad(true); in step()
182 torch::AutoGradMode enable_grad(true); in _directional_evaluate()446 torch::AutoGradMode enable_grad(true); in step()581 torch::AutoGradMode enable_grad(true); in step()
70 at::AutoGradMode enable_grad(true); in step()
77 at::AutoGradMode enable_grad(true); in step()
166 std::optional<c10::AutoGradMode> grad_guard; in autogradBasedTransformSendToNext()
304 torch::autograd::AutoGradMode guard(false); in main()
97 AutoGradMode grad_mode(false); in accumulateGrad()