Searched defs:enable_grad (Results 1 – 13 of 13) sorted by relevance
182 torch::AutoGradMode enable_grad(true); in _directional_evaluate() local446 torch::AutoGradMode enable_grad(true); in step() local581 torch::AutoGradMode enable_grad(true); in step() local
66 at::AutoGradMode enable_grad(true); in step() local
77 at::AutoGradMode enable_grad(true); in step() local
70 at::AutoGradMode enable_grad(true); in step() local
860 at::AutoGradMode enable_grad(true); in TEST() local876 at::AutoGradMode enable_grad(true); in TEST() local896 at::AutoGradMode enable_grad(true); in TEST() local909 at::AutoGradMode enable_grad(true); in TEST() local939 at::AutoGradMode enable_grad(true); in TEST() local953 at::AutoGradMode enable_grad(true); in TEST() local
83 at::AutoGradMode enable_grad(true); in step() local
44 def __call__(self, enable_grad, wrapped_func, *args, **kwargs): argument
88 class enable_grad(_NoParamDecoratorContextManager): class
345 enable_grad=None, argument
2053 enable_grad as enable_grad, unknown
3187 def enable_grad(): function