|
tvm
|
The set of Relay analysis passes written in C++. More...
#include <tvm/ir/module.h>#include <tvm/relay/adt.h>#include <tvm/relay/expr.h>#include <tvm/relay/function.h>#include <tvm/relay/type.h>#include <tvm/runtime/logging.h>#include <string>#include <unordered_map>Go to the source code of this file.
Namespaces | |
| tvm | |
| runtime implementation for LibTorch/TorchScript. | |
| tvm::relay | |
| Relay: a high level functional IR for TVM. | |
Functions | |
| Kind | tvm::relay::KindCheck (const Type &t, const IRModule &mod, Optional< DiagnosticContext > diag_ctx=Optional< DiagnosticContext >()) |
| Check that types are well kinded by applying "kinding rules". More... | |
| bool | tvm::relay::ConstantCheck (const Expr &e) |
| Check whether an expression is constant. More... | |
| bool | tvm::relay::BasicBlockNormalFormCheck (const Expr &e) |
| Check whether an expression is in the basic block normal form. More... | |
| bool | tvm::relay::WellFormed (const Expr &expr, Optional< DiagnosticContext > diag_ctx=Optional< DiagnosticContext >()) |
| Check that each Var is only bound once. More... | |
| tvm::Array< Var > | tvm::relay::BoundVars (const Expr &expr) |
| Get all bound variables from expression expr. More... | |
| tvm::Array< Var > | tvm::relay::BoundVars (const Pattern &pat) |
| Get all bound variables from pattern pat. More... | |
| tvm::Array< Var > | tvm::relay::FreeVars (const Expr &expr) |
| Get free type parameters from expression expr. More... | |
| tvm::Array< Var > | tvm::relay::AllVars (const Expr &expr) |
| Get all variables from expression expr. More... | |
| tvm::Array< TypeVar > | tvm::relay::FreeTypeVars (const Expr &expr, const IRModule &mod) |
| Get free TypeVars from expression expr. More... | |
| tvm::Array< TypeVar > | tvm::relay::FreeTypeVars (const Type &t, const IRModule &mod) |
| Get free TypeVars from type t. More... | |
| tvm::Array< TypeVar > | tvm::relay::BoundTypeVars (const Expr &expr, const IRModule &mod) |
| Get all bound type variables from expression expr. More... | |
| tvm::Array< TypeVar > | tvm::relay::BoundTypeVars (const Type &t, const IRModule &mod) |
| Get all bound type variables from type t. More... | |
| tvm::Array< TypeVar > | tvm::relay::AllTypeVars (const Expr &expr, const IRModule &mod) |
| Get all type variables in expression expr. More... | |
| tvm::Array< TypeVar > | tvm::relay::AllTypeVars (const Type &t, const IRModule &mod) |
| Get all type variables in type t. More... | |
| Array< Pattern > | tvm::relay::UnmatchedCases (const Match &match, const IRModule &mod) |
| Finds cases that the given match expression does not catch, if any. More... | |
| std::unordered_map< const Object *, size_t > | tvm::relay::GetExprRefCount (const Expr &body) |
| Get reference counter of each internal ExprNode in body. More... | |
| IRModule | tvm::relay::GetCalibrateModule (IRModule mod) |
| Get the updated module for collecting calibration data. More... | |
| Map< GlobalVar, Array< Integer > > | tvm::relay::GetCalibrateOutputMap (const IRModule &mod) |
| Get the output map between subgrpahs and its inputs/output. More... | |
The set of Relay analysis passes written in C++.
1.8.13