ndarray.h File Reference

A device-independent managed NDArray abstraction. More...

#include <tvm/runtime/c_runtime_api.h>
#include <tvm/runtime/container/optional.h>
#include <tvm/runtime/container/shape_tuple.h>
#include <tvm/runtime/container/string.h>
#include <tvm/runtime/data_type.h>
#include <tvm/runtime/object.h>
#include <tvm/runtime/serializer.h>
#include <atomic>
#include <functional>
#include <utility>
#include <vector>

Include dependency graph for ndarray.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::runtime::NDArray
	Managed NDArray. The array is backed by reference counted blocks. More...
class	tvm::runtime::NDArray::ContainerBase
	The container base structure contains all the fields except for the Object header. More...
class	tvm::runtime::NDArray::Container
	Object container class that backs NDArray. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::runtime

Typedefs
using	tvm::Device = DLDevice

Functions
bool	tvm::runtime::SaveDLTensor (dmlc::Stream *strm, const DLTensor *tensor)
	Save a DLTensor to stream. More...
size_t	tvm::runtime::GetDataSize (const DLTensor &arr)
	return the size of data the DLTensor hold, in term of number of bytes More...
Object *	tvm::runtime::TVMArrayHandleToObjectHandle (TVMArrayHandle handle)
Device	tvm::runtime::GetPreferredHostDevice (Device device)
	Get the preferred host device from the input device. More...

Variables
constexpr uint64_t	tvm::runtime::kTVMNDArrayMagic = 0xDD5E40F096B4A13F
	Magic number for NDArray file. More...

Detailed Description

A device-independent managed NDArray abstraction.