tensor.h File Reference

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

#include <tvm/ffi/container/shape.h>
#include <tvm/ffi/container/tensor.h>
#include <tvm/ffi/dtype.h>
#include <tvm/ffi/optional.h>
#include <tvm/ffi/string.h>
#include <tvm/runtime/base.h>
#include <tvm/runtime/device_api.h>
#include <tvm/support/io.h>
#include <tvm/support/serializer.h>
#include <atomic>
#include <functional>
#include <utility>
#include <vector>

Include dependency graph for tensor.h:

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

Go to the source code of this file.

Classes
class	tvm::runtime::Tensor
	Managed Tensor. The array is backed by reference counted blocks. More...

Namespaces
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::runtime

Functions
bool	tvm::runtime::SaveDLTensor (support::Stream *strm, const DLTensor *tensor)
	Save a DLTensor to stream.
Device	tvm::runtime::GetPreferredHostDevice (Device device)
	Get the preferred host device from the input device.

Variables
constexpr uint64_t	tvm::runtime::kTVMTensorMagic = 0xDD5E40F096B4A13F
	Magic number for Tensor file.

Detailed Description

A device-independent managed Tensor abstraction.