Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
at::native::TensorDescriptor Class Reference
Inheritance diagram for at::native::TensorDescriptor:
at::native::Descriptor< miopenTensorDescriptor,&miopenCreateTensorDescriptor,&miopenDestroyTensorDescriptor > at::native::Descriptor< cudnnTensorStruct,&cudnnCreateTensorDescriptor,&cudnnDestroyTensorDescriptor >

Public Member Functions

 TensorDescriptor (const at::Tensor &t, size_t pad=0)
 
void set (const at::Tensor &t, size_t pad=0)
 
void set (cudnnDataType_t dataType, IntArrayRef sizes, IntArrayRef strides, size_t pad=0)
 
void print ()
 
 TensorDescriptor (const at::Tensor &t, size_t pad=0)
 
void set (const at::Tensor &t, size_t pad=0)
 
void set (miopenDataType_t dataType, IntArrayRef sizes, IntArrayRef strides, size_t pad=0)
 
void print ()
 
- Public Member Functions inherited from at::native::Descriptor< miopenTensorDescriptor,&miopenCreateTensorDescriptor,&miopenDestroyTensorDescriptor >
miopenTensorDescriptor * desc () const
 
miopenTensorDescriptor * desc ()
 
miopenTensorDescriptor * desc () const
 
miopenTensorDescriptor * desc ()
 
miopenTensorDescriptor * mut_desc ()
 
miopenTensorDescriptor * mut_desc ()
 
- Public Member Functions inherited from at::native::Descriptor< cudnnTensorStruct,&cudnnCreateTensorDescriptor,&cudnnDestroyTensorDescriptor >
cudnnTensorStruct * desc () const
 
cudnnTensorStruct * desc ()
 
cudnnTensorStruct * desc () const
 
cudnnTensorStruct * desc ()
 
cudnnTensorStruct * mut_desc ()
 
cudnnTensorStruct * mut_desc ()
 

Additional Inherited Members

- Protected Member Functions inherited from at::native::Descriptor< miopenTensorDescriptor,&miopenCreateTensorDescriptor,&miopenDestroyTensorDescriptor >
void init ()
 
void init ()
 
- Protected Member Functions inherited from at::native::Descriptor< cudnnTensorStruct,&cudnnCreateTensorDescriptor,&cudnnDestroyTensorDescriptor >
void init ()
 
void init ()
 

Detailed Description

Definition at line 95 of file Descriptors.h.


The documentation for this class was generated from the following files: