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