1 | #pragma once |
---|---|
2 | #include <c10/core/QScheme.h> |
3 | |
4 | // Forward declarations of core ATen types used in dispatch functions |
5 | namespace c10 { |
6 | |
7 | template<typename T> |
8 | class optional; |
9 | template<typename T> |
10 | class List; |
11 | template<typename T> |
12 | class IListRef; |
13 | class Stream; |
14 | class Scalar; |
15 | class SymInt; |
16 | class SymIntList; |
17 | struct Storage; |
18 | struct TensorOptions; |
19 | template <typename T> |
20 | class ArrayRef; |
21 | template <typename T> |
22 | class OptionalArrayRef; |
23 | |
24 | } // namespace c10 |
25 | |
26 | namespace at { |
27 | |
28 | class Tensor; |
29 | class OptionalTensorRef; |
30 | struct Dimname; |
31 | struct Generator; |
32 | using TensorList = c10::ArrayRef<Tensor>; |
33 | using ITensorListRef = c10::IListRef<Tensor>; |
34 | using IOptTensorListRef = c10::IListRef<OptionalTensorRef>; |
35 | using DimnameList = c10::ArrayRef<Dimname>; |
36 | using IntArrayRef = c10::ArrayRef<int64_t>; |
37 | using OptionalIntArrayRef = c10::OptionalArrayRef<int64_t>; |
38 | using OptionalSymIntArrayRef = c10::OptionalArrayRef<c10::SymInt>; |
39 | |
40 | using c10::Stream; |
41 | using c10::Storage; |
42 | using c10::QScheme; |
43 | using c10::Scalar; |
44 | using c10::SymInt; |
45 | using c10::SymIntList; |
46 | using c10::TensorOptions; |
47 | |
48 | } // namespace at |
49 |