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

Public Types

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

Public Member Functions

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

Additional Inherited Members

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

Definition at line 159 of file TensorAccessor.h.


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