Caffe2 - C++ API
A deep learning, cross platform ML framework
python_variable_methods_dispatch.h
1 #pragma once
2 
3 // ${generated_comment}
4 
5 #include "torch/csrc/utils/auto_gil.h"
6 
7 #include <ATen/ATen.h>
8 
9 // Contains inline wrappers around ATen functions that release the GIL and
10 // switch to the correct CUDA device.
11 
12 namespace torch { namespace autograd {
13 
14 using at::Tensor;
15 using at::Scalar;
16 using at::TensorList;
17 using at::IntArrayRef;
18 using at::Generator;
20 using at::Storage;
21 
22 ${py_method_dispatch}
23 
24 }} // namespace torch::autograd
Scalar represents a 0-dimensional tensor which contains a single element.
Definition: Scalar.h:22
Definition: jit_type.h:17
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory)...
Definition: ArrayRef.h:41