|
typedef PtrTraits< T >::PtrType | PtrType |
|
typedef PtrTraits< T >::PtrType | PtrType |
|
|
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 |
|
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 |
|
|
PtrType | data_ |
|
index_t | sizes_ [N] |
|
index_t | strides_ [N] |
|
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: