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

Public Types

typedef PtrTraits< T >::PtrType PtrType
 
- Public Types inherited from at::TensorAccessorBase< T, N, PtrTraits, index_t >
typedef PtrTraits< T >::PtrType PtrType
 

Public Member Functions

C10_HOST_DEVICE TensorAccessor (PtrType data_, const index_t *sizes_, const index_t *strides_)
 
C10_HOST_DEVICE TensorAccessor< T, N-1, PtrTraits, index_t > operator[] (index_t i)
 
C10_HOST_DEVICE const TensorAccessor< T, N-1, PtrTraits, index_t > operator[] (index_t i) const
 
- Public Member Functions inherited from at::TensorAccessorBase< T, N, PtrTraits, index_t >
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
 

Additional Inherited Members

- Protected Attributes inherited from at::TensorAccessorBase< T, N, PtrTraits, index_t >
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::TensorAccessor< T, N, PtrTraits, index_t >

Definition at line 68 of file TensorAccessor.h.


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