tvm
|
A pattern matcher for matching dataflow properties. More...
#include <tvm/relay/dataflow_pattern.h>
#include <tvm/relay/dataflow_pattern_functor.h>
#include <string>
#include <unordered_map>
#include <utility>
Go to the source code of this file.
Classes | |
class | tvm::relay::DFPatternCallbackNode |
Base type of all dataflow pattern callbacks. More... | |
class | tvm::relay::DFPatternCallback |
Managed reference to dataflow pattern callbacks. More... | |
Namespaces | |
tvm | |
runtime implementation for LibTorch/TorchScript. | |
tvm::relay | |
Relay: a high level functional IR for TVM. | |
Functions | |
bool | tvm::relay::MatchPattern (DFPattern pattern, Expr expr) |
Determine if a pattern matches an expression. More... | |
Expr | tvm::relay::RewritePatterns (Array< DFPatternCallback > callbacks, Expr expr, IRModule mod=IRModule()) |
Rewrite an expression based on some number of DFPatternCallbacks. More... | |
Expr | tvm::relay::PartitionPattern (DFPattern pattern, Expr expr, Map< String, ObjectRef > attrs, PackedFunc check) |
Partition all matches of a DFPattern inside an Expr into separate Function calls. More... | |
Expr | tvm::relay::InferType (const Expr &expr) |
Infer the type of an expression. More... | |
A pattern matcher for matching dataflow properties.