xref: /aosp_15_r20/external/pytorch/tools/autograd/templates/ViewFuncs.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker #pragma once
2*da0073e9SAndroid Build Coastguard Worker 
3*da0073e9SAndroid Build Coastguard Worker // ${generated_comment}
4*da0073e9SAndroid Build Coastguard Worker 
5*da0073e9SAndroid Build Coastguard Worker #include <torch/library.h>
6*da0073e9SAndroid Build Coastguard Worker #include <torch/csrc/autograd/variable.h>
7*da0073e9SAndroid Build Coastguard Worker #include <c10/core/SymIntArrayRef.h>
8*da0073e9SAndroid Build Coastguard Worker 
9*da0073e9SAndroid Build Coastguard Worker #ifndef AT_PER_OPERATOR_HEADERS
10*da0073e9SAndroid Build Coastguard Worker #include <ATen/Operators.h>
11*da0073e9SAndroid Build Coastguard Worker #else
12*da0073e9SAndroid Build Coastguard Worker $ops_headers
13*da0073e9SAndroid Build Coastguard Worker #endif
14*da0073e9SAndroid Build Coastguard Worker 
15*da0073e9SAndroid Build Coastguard Worker namespace torch::autograd::generated {
16*da0073e9SAndroid Build Coastguard Worker 
17*da0073e9SAndroid Build Coastguard Worker using at::Scalar;
18*da0073e9SAndroid Build Coastguard Worker using at::Tensor;
19*da0073e9SAndroid Build Coastguard Worker using at::IntArrayRef;
20*da0073e9SAndroid Build Coastguard Worker using at::ArrayRef;
21*da0073e9SAndroid Build Coastguard Worker using at::Type;
22*da0073e9SAndroid Build Coastguard Worker using at::ScalarType;
23*da0073e9SAndroid Build Coastguard Worker using std::optional;
24*da0073e9SAndroid Build Coastguard Worker using c10::fmap;
25*da0073e9SAndroid Build Coastguard Worker 
26*da0073e9SAndroid Build Coastguard Worker ${view_func_declarations}
27*da0073e9SAndroid Build Coastguard Worker 
28*da0073e9SAndroid Build Coastguard Worker } // namespace torch::autograd::generated
29