ndarray.h File Reference

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

#include <tvm/ffi/container/ndarray.h>
#include <tvm/ffi/container/shape.h>
#include <tvm/ffi/optional.h>
#include <tvm/ffi/string.h>
#include <tvm/runtime/base.h>
#include <tvm/runtime/data_type.h>
#include <tvm/runtime/device_api.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...

Namespaces
	tvm
	Performance counters for profiling via the PAPI library.
	tvm::runtime

Functions
bool	tvm::runtime::SaveDLTensor (dmlc::Stream *strm, const DLTensor *tensor)
	Save a DLTensor to stream. More...
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.