1 #pragma once 2 3 #include <ATen/core/grad_mode.h> 4 #include <torch/csrc/Export.h> 5 6 namespace torch::autograd { 7 8 using GradMode = at::GradMode; 9 using AutoGradMode = at::AutoGradMode; 10 11 } // namespace torch::autograd 12