tvm
|
Attributes for memory operators. More...
#include <tvm/ir/attrs.h>
#include <tvm/relay/expr.h>
#include <tvm/target/virtual_device.h>
#include <string>
#include <vector>
Go to the source code of this file.
Classes | |
struct | tvm::relay::AllocStorageAttrs |
Options for allocating storage. More... | |
struct | tvm::relay::AllocTensorAttrs |
Options for allocating tensors. More... | |
Namespaces | |
tvm | |
runtime implementation for LibTorch/TorchScript. | |
tvm::relay | |
Relay: a high level functional IR for TVM. | |
Functions | |
std::vector< TensorType > | tvm::relay::FlattenTupleType (const Type &type) |
std::vector< Expr > | tvm::relay::FromTupleType (const Type &type, const Expr &expr) |
Expr | tvm::relay::ToTupleType (const Type &t, const std::vector< Expr > &exprs) |
Attributes for memory operators.