|
tvm
|
Relax types, including the richer dependent Relax type nodes. More...
#include <tvm/ffi/function.h>#include <tvm/ffi/reflection/registry.h>#include <tvm/ir/attrs.h>#include <tvm/ir/env_func.h>#include <tvm/ir/global_info.h>#include <tvm/ir/type.h>#include <tvm/tirx/expr.h>#include <string>#include <utility>Go to the source code of this file.
Classes | |
| class | tvm::relax::PackedFuncTypeNode |
| class | tvm::relax::PackedFuncType |
| class | tvm::relax::AnyTypeNode |
| Base type of all Relax type information. More... | |
| class | tvm::relax::AnyType |
| Managed reference to AnyTypeNode. More... | |
| class | tvm::relax::ShapeTypeNode |
| Type of shape value. More... | |
| class | tvm::relax::ShapeType |
| Managed reference to ShapeTypeNode. More... | |
| class | tvm::relax::TensorTypeNode |
| Type of Tensor. More... | |
| class | tvm::relax::TensorType |
| Managed reference to TensorTypeNode. More... | |
| class | tvm::relax::FuncTypeNode |
| Function type information. More... | |
| class | tvm::relax::FuncType |
| Managed reference to FuncTypeNode. More... | |
Namespaces | |
| tvm | |
| An object that builds and maintains block scope and StmtSref mapping for Dependence analysis. | |
| tvm::relax | |
Typedefs | |
| using | tvm::relax::Expr = tvm::Expr |
| using | tvm::relax::ExprNode = tvm::ExprNode |
| using | tvm::relax::ObjectTypeNode = AnyTypeNode |
| using | tvm::relax::ObjectType = AnyType |
| using | tvm::relax::TypeDeriveFunc = TypedEnvFunc< Type(const Call &call, const BlockBuilder &ctx)> |
| custom-defined Type derivation function. More... | |
Functions | |
| template<typename T > | |
| ffi::Optional< T > | tvm::relax::MatchType (const Expr &expr) |
| Match and check if expr has Relax type T and return it. More... | |
| template<typename T > | |
| const T * | tvm::relax::GetTypeAs (const Expr &expr) |
| Get the type of a given expr and try to cast it as const T*. More... | |
| Type | tvm::relax::GetType (const Expr &expr) |
| Get the underlying Relax type of expr. More... | |
| bool | tvm::relax::HasVoidType (const Expr &expr) |
| Whether the expr has void type. More... | |
| void | tvm::relax::UpdateType (Expr expr, Type ty) |
| Update the type of an Expr. More... | |
Relax types, including the richer dependent Relax type nodes.