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: