Tensor data structure used by custom operator. More...

#include <lib_api.h>

Collaboration diagram for MXTensor:

Public Member Functions
	MXTensor ()

	MXTensor (const MXTensor &oth)

	MXTensor (void *data_ptr, const std::vector< int64_t > &shape, MXDType dtype, size_t vID, MXContext mx_ctx, MXStorageType stype=kDefaultStorage)

void	setTensor (void dptr, MXDType type, const int64_t dims, int ndims, size_t vID, MXContext mx_ctx, MXStorageType storage_type)
	populate internal tensor fields More...

void	setDLTensor ()
	populate DLTensor fields More...

template<typename data_type >
data_type *	data ()
	helper function to cast data pointer More...

int64_t	size () const
	helper function to get data size More...

bool	isSame (const MXTensor &oth) const
	helper function to compare two MXTensors More...

Detailed Description

Tensor data structure used by custom operator.

Constructor & Destructor Documentation

MXTensor::MXTensor ( )

inline

MXTensor::MXTensor ( const MXTensor & oth )

inline

inline

template<typename data_type >

data_type* MXTensor::data ( )

inline

helper function to cast data pointer

bool MXTensor::isSame ( const MXTensor & oth ) const

inline

helper function to compare two MXTensors

void MXTensor::setDLTensor ( )

inline

populate DLTensor fields

inline

populate internal tensor fields

int64_t MXTensor::size ( ) const

inline

helper function to get data size

MXContext MXTensor::ctx

void* MXTensor::data_ptr

DLTensor MXTensor::dltensor

MXDType MXTensor::dtype

std::vector<int64_t> MXTensor::shape

MXStorageType MXTensor::stype

size_t MXTensor::verID

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