mxnet::ext::MXTensor Struct Reference

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

#include <lib_api.h>

Collaboration diagram for mxnet::ext::MXTensor:

Public Member Functions
	MXTensor ()
	MXTensor (const MXTensor &oth)
	MXTensor (void *data_ptr, 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...

Public Attributes
void *	data_ptr
std::vector< int64_t >	shape
MXDType	dtype
size_t	verID
MXContext	ctx
DLTensor	dltensor
MXStorageType	stype

Detailed Description

Tensor data structure used by custom operator.

Constructor & Destructor Documentation

MXTensor() [1/3]

mxnet::ext::MXTensor::MXTensor

(

)

MXTensor() [2/3]

mxnet::ext::MXTensor::MXTensor

(

const MXTensor &

oth

)

MXTensor() [3/3]

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

Member Function Documentation

data()

template<typename data_type >

data_type* mxnet::ext::MXTensor::data ( )

inline

helper function to cast data pointer

isSame()

bool mxnet::ext::MXTensor::isSame

(

const MXTensor &

oth

)

const

helper function to compare two MXTensors

setDLTensor()

void mxnet::ext::MXTensor::setDLTensor

(

)

populate DLTensor fields

setTensor()

void mxnet::ext::MXTensor::setTensor	(	void *	dptr,
		MXDType	type,
		const int64_t *	dims,
		int	ndims,
		size_t	vID,
		MXContext	mx_ctx,
		MXStorageType	storage_type
	)

populate internal tensor fields

size()

int64_t mxnet::ext::MXTensor::size

(

)

const

helper function to get data size

Member Data Documentation

ctx

MXContext mxnet::ext::MXTensor::ctx

data_ptr

void* mxnet::ext::MXTensor::data_ptr

dltensor

DLTensor mxnet::ext::MXTensor::dltensor

dtype

MXDType mxnet::ext::MXTensor::dtype

shape

std::vector<int64_t> mxnet::ext::MXTensor::shape

stype

MXStorageType mxnet::ext::MXTensor::stype

verID

size_t mxnet::ext::MXTensor::verID

---

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

• /work/mxnet/include/mxnet/lib_api.h