Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
tools
autograd
templates
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
;
19
using
at::SparseTensorRef
;
20
using
at::Storage
;
21
22
${py_method_dispatch}
23
24
}}
// namespace torch::autograd
at::Generator
Definition:
Generator.h:8
at::SparseTensorRef
Definition:
SparseTensorRef.h:6
at::Tensor
Definition:
Tensor.h:48
c10::Scalar
Scalar represents a 0-dimensional tensor which contains a single element.
Definition:
Scalar.h:22
torch
Definition:
jit_type.h:17
c10::ArrayRef
ArrayRef - Represent a constant reference to an array (0 or more elements consecutively in memory)...
Definition:
ArrayRef.h:41
c10::Storage
Definition:
Storage.h:7
Generated on Thu Mar 21 2019 13:06:22 for Caffe2 - C++ API by
1.8.11