Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Data Fields
torch::jit::fuser::TensorDesc Struct Reference

Public Member Functions

 TensorDesc (const at::ScalarType &type, const std::vector< bool > &contiguity)
 TensorDesc (const at::ScalarType &type, const at::IntArrayRef &sizes, const at::IntArrayRef &strides)
 TensorDesc (const at::Tensor &t)
 TensorDesc (const c10::CompleteTensorTypePtr &type)
size_t nDim () const
bool lastIsContiguous () const
bool operator== (const TensorDesc &desc) const
bool operator!= (const TensorDesc &desc) const

Static Public Member Functions

static std::vector< bool > findContiguous (const at::IntArrayRef &sizes, const at::IntArrayRef &strides)
static size_t hash (const TensorDesc &spec)

Data Fields

at::ScalarType scalar_type
std::vector< bool > contiguity

Detailed Description

Definition at line 20 of file tensor_desc.h.

The documentation for this struct was generated from the following file: