tvm
|
The Relay operator Strategy and related data structure. More...
#include <tvm/relay/expr.h>
#include <tvm/relay/op_attr_types.h>
#include <tvm/target/target.h>
#include <tvm/te/schedule.h>
#include <tvm/te/tensor.h>
#include <string>
Go to the source code of this file.
Classes | |
class | tvm::relay::OpImplementationNode |
Operator implementation that includes compute and schedule function. More... | |
class | tvm::relay::OpImplementation |
Operator implementation class. More... | |
class | tvm::relay::OpSpecializationNode |
Specialized implementations for operators under certain conditions. More... | |
class | tvm::relay::OpSpecialization |
Operator specialization class. More... | |
class | tvm::relay::OpStrategyNode |
Operator strategy to choose implementation. More... | |
class | tvm::relay::OpStrategy |
Operator strategy class. More... | |
Namespaces | |
tvm | |
runtime implementation for LibTorch/TorchScript. | |
tvm::relay | |
Relay: a high level functional IR for TVM. | |
The Relay operator Strategy and related data structure.