tvm
Classes | Namespaces | Functions
dataflow_matcher.h File Reference

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>
Include dependency graph for dataflow_matcher.h:

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...
 

Detailed Description

A pattern matcher for matching dataflow properties.