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: