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

Public Types

typedef PtrTraits< T >::PtrType PtrType
 

Public Member Functions

C10_HOST PackedTensorAccessorBase (PtrType data_, const index_t *sizes_, const index_t *strides_)
 
template<typename source_index_t , class = typename std::enable_if<std::is_same<source_index_t, int64_t>::value>::type>
C10_HOST PackedTensorAccessorBase (PtrType data_, const source_index_t *sizes_, const source_index_t *strides_)
 
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_
 
index_t sizes_ [N]
 
index_t strides_ [N]
 

Detailed Description

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

Definition at line 115 of file TensorAccessor.h.


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