cloudViewer::core::TensorRef Struct Reference

A minimalistic class that reference a Tensor. More...

#include <Indexer.h>

Collaboration diagram for cloudViewer::core::TensorRef:

Public Member Functions
	TensorRef ()
	TensorRef (const Tensor &t)
void	Permute (const SizeVector &dims)
	Permute (dimension shuffle) the reference to a Tensor. More...
bool	IsContiguous () const
	Returns True if the underlying memory buffer is contiguous. More...
bool	operator== (const TensorRef &other) const
bool	operator!= (const TensorRef &other) const

Public Attributes
void *	data_ptr_
int64_t	ndims_ = 0
int64_t	dtype_byte_size_ = 0
int64_t	shape_ [MAX_DIMS]
int64_t	byte_strides_ [MAX_DIMS]

Detailed Description

A minimalistic class that reference a Tensor.

Definition at line 106 of file Indexer.h.

Constructor & Destructor Documentation

TensorRef() [1/2]

cloudViewer::core::TensorRef::TensorRef ( )

inline

Definition at line 110 of file Indexer.h.

TensorRef() [2/2]

cloudViewer::core::TensorRef::TensorRef ( const Tensor &  t )

inline

Definition at line 112 of file Indexer.h.

References byte_strides_, cloudViewer::core::Dtype::ByteSize(), data_ptr_, dtype_byte_size_, cloudViewer::core::Tensor::GetDataPtr(), cloudViewer::core::Tensor::GetDtype(), cloudViewer::core::Tensor::GetShape(), cloudViewer::core::Tensor::GetStride(), LogError, cloudViewer::core::MAX_DIMS, ndims_, cloudViewer::core::Tensor::NumDims(), and shape_.

Member Function Documentation

IsContiguous()

bool cloudViewer::core::TensorRef::IsContiguous ( ) const

inline

Returns True if the underlying memory buffer is contiguous.

Definition at line 164 of file Indexer.h.

References byte_strides_, cloudViewer::core::shape_util::DefaultStrides(), dtype_byte_size_, ndims_, and shape_.

Referenced by cloudViewer::core::Indexer::UpdateContiguousFlags().

operator!=()

bool cloudViewer::core::TensorRef::operator!= ( const TensorRef &  other ) const

inline

Definition at line 186 of file Indexer.h.

operator==()

bool cloudViewer::core::TensorRef::operator== ( const TensorRef &  other ) const

inline

Definition at line 174 of file Indexer.h.

References byte_strides_, data_ptr_, dtype_byte_size_, ndims_, and shape_.

Permute()

void cloudViewer::core::TensorRef::Permute ( const SizeVector &  dims )

inline

Permute (dimension shuffle) the reference to a Tensor.

Parameters

dims	The desired ordering of dimensions.

Note: This only affects this Tensor reference, but not the underlying Tensor.

Definition at line 132 of file Indexer.h.

References byte_strides_, LogError, ndims_, shape_, cloudViewer::core::SmallVectorBase< Size_T >::size(), and cloudViewer::core::shape_util::WrapDim().

Referenced by cloudViewer::core::Indexer::ReorderDimensions().

Member Data Documentation

byte_strides_

int64_t cloudViewer::core::TensorRef::byte_strides_[MAX_DIMS]

Definition at line 197 of file Indexer.h.

Referenced by cloudViewer::core::Indexer::BroadcastRestride(), cloudViewer::core::Indexer::CoalesceDimensions(), cloudViewer::core::Indexer::GetPerOutputIndexer(), cloudViewer::core::TensorIterator::GetPtr(), cloudViewer::core::Indexer::GetWorkloadDataPtr(), IsContiguous(), cloudViewer::core::Indexer::IsReductionDim(), operator==(), Permute(), cloudViewer::core::Indexer::ReductionRestride(), cloudViewer::core::Indexer::ReorderDimensions(), and TensorRef().

data_ptr_

void* cloudViewer::core::TensorRef::data_ptr_

Definition at line 193 of file Indexer.h.

Referenced by cloudViewer::core::Indexer::GetPerOutputIndexer(), cloudViewer::core::TensorIterator::GetPtr(), cloudViewer::core::Indexer::GetWorkloadDataPtr(), operator==(), cloudViewer::core::Indexer::ShrinkDim(), and TensorRef().

dtype_byte_size_

int64_t cloudViewer::core::TensorRef::dtype_byte_size_ = 0

Definition at line 195 of file Indexer.h.

Referenced by cloudViewer::core::TensorIterator::GetPtr(), cloudViewer::core::Indexer::GetWorkloadDataPtr(), IsContiguous(), operator==(), and TensorRef().

ndims_

int64_t cloudViewer::core::TensorRef::ndims_ = 0

Definition at line 194 of file Indexer.h.

Referenced by cloudViewer::core::Indexer::BroadcastRestride(), cloudViewer::core::Indexer::CoalesceDimensions(), cloudViewer::core::Indexer::Indexer(), IsContiguous(), operator==(), Permute(), cloudViewer::core::Indexer::ReductionRestride(), and TensorRef().

shape_

int64_t cloudViewer::core::TensorRef::shape_[MAX_DIMS]

Definition at line 196 of file Indexer.h.

Referenced by cloudViewer::core::Indexer::BroadcastRestride(), cloudViewer::core::Indexer::GetPerOutputIndexer(), cloudViewer::core::Indexer::Indexer(), IsContiguous(), cloudViewer::core::TensorIterator::NumWorkloads(), operator==(), Permute(), cloudViewer::core::Indexer::ReductionRestride(), and TensorRef().

---

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

• /root/ACloudViewer/libs/cloudViewer/core/Indexer.h