|
tvm
|
A device-independent managed Tensor abstraction. More...
#include <tvm/ffi/container/shape.h>#include <tvm/ffi/container/tensor.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>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 | |
| 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::kTVMTensorMagic = 0xDD5E40F096B4A13F |
| Magic number for Tensor file. More... | |
A device-independent managed Tensor abstraction.