PyTorch
Loading...
Searching...
No Matches
ATen.h
Go to the documentation of this file.
1#pragma once
2
3#if !defined(_MSC_VER) && __cplusplus < 201402L
4#error C++14 or later compatible compiler is required to use ATen.
5#endif
6
7#include <ATen/Context.h>
8#include <ATen/Device.h>
9#include <ATen/DeviceGuard.h>
10#include <ATen/DimVector.h>
11#include <ATen/Dispatch.h>
12#include <ATen/Formatting.h>
13#include <ATen/Functions.h>
14#include <ATen/NamedTensor.h>
15#include <ATen/ScalarOps.h>
16#include <ATen/Tensor.h>
17#include <ATen/TensorGeometry.h>
18#include <ATen/TensorIndexing.h>
19#include <ATen/TensorOperators.h>
20#include <ATen/Version.h>
21#include <ATen/core/ATenGeneral.h>
22#include <ATen/core/Generator.h>
23#include <ATen/core/Reduction.h>
24#include <ATen/core/Scalar.h>
25#include <ATen/core/UnsafeFromTH.h>
26#include <ATen/core/ivalue.h>
27#include <ATen/core/jit_type.h>
28#include <c10/core/Allocator.h>
29#include <c10/core/InferenceMode.h>
30#include <c10/core/Layout.h>
31#include <c10/core/Storage.h>
32#include <c10/core/TensorOptions.h>
33#include <c10/util/Exception.h>
34
35// TODO: try to remove this
36// There is some back story, see https://github.com/pytorch/pytorch/issues/48684
37#include <ATen/NativeFunctions.h>