|
typedef PtrTraits< T >::PtrType | PtrType |
|
typedef PtrTraits< T >::PtrType | PtrType |
|
|
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 |
|
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 |
|
|
PtrType | data_ |
|
const index_t * | sizes_ |
|
const index_t * | strides_ |
|
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: