|
tvm
|
#include <tvm/ffi/container/array.h>#include <tvm/ffi/container/map.h>#include <tvm/ffi/reflection/registry.h>#include <tvm/ir/cow.h>#include <tvm/ir/expr.h>#include <tvm/ir/function.h>#include <tvm/ir/source_map.h>#include <tvm/relax/type.h>#include <tvm/runtime/tensor.h>#include <tvm/tirx/expr.h>#include <tvm/tirx/op.h>#include <functional>Go to the source code of this file.
Namespaces | |
| tvm | |
| An object that builds and maintains block scope and StmtSref mapping for Dependence analysis. | |
| tvm::relax | |
| tvm::relax::attr | |
Functions | |
| Expr | tvm::relax::GetShapeOf (const Expr &expr) |
| Get the shape of Expr. More... | |
Variables | |
| constexpr const char * | tvm::relax::attr::kPrimitive = "Primitive" |
| Mark the function as a primitive function. More... | |
| constexpr const char * | tvm::relax::attr::kCodegen = "Codegen" |
| Indicate the codegen that should be used for building this function. When this is unset or set to "default", the default compilation pipeline will be used. More... | |
| constexpr const char * | tvm::relax::attr::kComposite = "Composite" |
| Treat the function as a composite operator. More... | |
| constexpr const char * | tvm::relax::attr::kPartitionedFromPattern = "PartitionedFromPattern" |
| Indicate the function was created by the Pattern Partitioning Pass. More... | |
| constexpr const char * | tvm::relax::attr::kWorkspaceSize = "WorkspaceSize" |
| The required workspace for an external function. More... | |
| constexpr const char * | tvm::relax::attr::kForcePure = "relax.force_pure" |
| Override checking purity for this function and treat as pure (is_pure must be set to true) More... | |
| constexpr const char * | tvm::relax::attr::kNumInput = "num_input" |
| The number of inputs of a function. If a function has the num_input attribute, the last func->params.size() - num_inputs arguments are assumed to be weights that are fixed across invocations. More... | |