tvm
|
A compile time representation for where data is to be stored at runtime, and how to compile code to compute it. More...
#include <tvm/ir/transform.h>
#include <tvm/target/target.h>
#include <string>
#include <unordered_set>
#include <utility>
Go to the source code of this file.
Classes | |
class | tvm::VirtualDeviceNode |
Describes at compile time the constraints on where data is to be stored at runtime down to the (virtual) device and memory scope level, and how to compile code to compute that data. Used by the PlanDevices pass to collect and solve (virtual) device constraints for the whole Relay program. More... | |
class | tvm::VirtualDevice |
Managed reference class to VirtualDeviceNode . More... | |
class | tvm::VirtualDeviceCache |
A cache of VirtualDevices . This can be used: More... | |
Namespaces | |
tvm | |
runtime implementation for LibTorch/TorchScript. | |
Variables | |
constexpr int | tvm::kNullDeviceType = 0 |
constexpr int | tvm::kInvalidDeviceType = -1 |
constexpr const char * | tvm::kVirtualDevice = "virtual_device" |
A compile time representation for where data is to be stored at runtime, and how to compile code to compute it.