Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Protected Attributes
at::TensorAccessorBase< T, N, PtrTraits, index_t > Class Template Reference
Inheritance diagram for at::TensorAccessorBase< T, N, PtrTraits, index_t >:
at::TensorAccessor< T, N, PtrTraits, index_t >

Public Types

typedef PtrTraits< T >::PtrType PtrType
 

Public Member Functions

C10_HOST_DEVICE TensorAccessorBase (PtrType data_, const index_t *sizes_, const index_t *strides_)
 
C10_HOST IntArrayRef sizes () const
 
C10_HOST IntArrayRef strides () const
 
C10_HOST_DEVICE index_t stride (index_t i) const
 
C10_HOST_DEVICE index_t size (index_t i) const
 
C10_HOST_DEVICE PtrType data ()
 
C10_HOST_DEVICE const PtrType data () const
 

Protected Attributes

PtrType data_
 
const index_t * sizes_
 
const index_t * strides_
 

Detailed Description

template<typename T, size_t N, template< typename U > class PtrTraits = DefaultPtrTraits, typename index_t = int64_t>
class at::TensorAccessorBase< T, N, PtrTraits, index_t >

Definition at line 30 of file TensorAccessor.h.

The documentation for this class was generated from the following file:
Generated on Thu Mar 21 2019 13:06:25 for Caffe2 - C++ API by   doxygen 1.8.11