24 #ifndef TVM_RELAX_DATAFLOW_PATTERN_H_
25 #define TVM_RELAX_DATAFLOW_PATTERN_H_
27 #include <tvm/ffi/container/array.h>
28 #include <tvm/ffi/optional.h>
29 #include <tvm/ffi/reflection/registry.h>
58 class StructInfoPattern;
59 class DataTypePattern;
61 class SameShapeConstraint;
104 template <
typename... Args>
197 std::function<ffi::Optional<Var>(
const DFPatternNode*)> match_state)
const = 0;
231 TVM_DLL
explicit PatternSeq(tvm::ffi::Array<DFPattern> patterns,
bool only_used_by =
false);
260 std::map<DFPattern, std::vector<std::pair<DFPattern, std::vector<PairCons>>>>
edge_constraints;
282 ICHECK(get() !=
nullptr);
287 ICHECK(get() !=
nullptr);
299 auto& pairs = (*this)->edge_constraints[producer];
300 auto it = std::find_if(pairs.begin(), pairs.end(),
301 [consumer](
auto p) { return p.first == consumer; });
302 if (it == pairs.end()) {
303 pairs.emplace_back(consumer, std::vector{cons});
305 auto& vec = it->second;
306 ICHECK(std::find(vec.cbegin(), vec.cend(), cons) == vec.cend())
307 <<
"Constraint already exists";
311 auto& patterns = (*this)->src_ordered;
312 if (std::find(patterns.begin(), patterns.end(), producer) == patterns.end()) {
313 patterns.push_back(producer);
323 (*this)->validation_constraints.push_back(constraint);
327 TVM_DLL
static ffi::Optional<PatternContext>
Current();
405 refl::ObjectDef<DataflowVarPatternNode>();
450 refl::ObjectDef<ConstantPatternNode>();
477 tvm::ffi::Array<DFPattern>
args;
490 refl::ObjectDef<CallPatternNode>()
547 refl::ObjectDef<FunctionPatternNode>()
606 refl::ObjectDef<UnorderedTuplePatternNode>().def_ro(
"fields",
636 refl::ObjectDef<TupleGetItemPatternNode>()
666 refl::ObjectDef<AndPatternNode>()
694 refl::ObjectDef<OrPatternNode>()
744 refl::ObjectDef<WildcardPatternNode>();
758 TVM_FFI_ICHECK(data !=
nullptr);
759 data_ = std::move(data);
781 refl::ObjectDef<StructInfoPatternNode>()
806 refl::ObjectDef<ShapePatternNode>()
834 std::function<ffi::Optional<Var>(
const DFPatternNode*)> match_state)
const override;
866 refl::ObjectDef<DataTypePatternNode>()
895 refl::ObjectDef<AttrPatternNode>()
926 refl::ObjectDef<ExternFuncPatternNode>().def_ro(
"global_symbol",
960 ffi::Optional<TuplePattern> args = std::nullopt);
969 template <
typename... Args>
971 return CallPattern(ffi::GetRef<DFPattern>(this->get()),
972 ffi::Array<DFPattern>({std::forward<Args>(args)...}));
Managed reference to DictAttrsNode.
Definition: attrs.h:162
Managed reference to RelaxExprNode.
Definition: expr.h:439
RAII wrapper function to enter and exit a context object similar to python's with syntax.
Definition: with.h:58
Match a conjunction of other patterns.
Definition: dataflow_pattern.h:659
DFPattern left
Definition: dataflow_pattern.h:661
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.AndPattern", AndPatternNode, DFPatternNode)
DFPattern right
Definition: dataflow_pattern.h:662
static void RegisterReflection()
Definition: dataflow_pattern.h:664
Managed reference to AndPatternNode.
Definition: dataflow_pattern.h:677
AndPattern(DFPattern lhs, DFPattern rhs)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(AndPattern, DFPattern, AndPatternNode)
A pattern that asserting a root pattern has certain attributes.
Definition: dataflow_pattern.h:888
DictAttrs attrs
Definition: dataflow_pattern.h:891
DFPattern pattern
Definition: dataflow_pattern.h:890
static void RegisterReflection()
Definition: dataflow_pattern.h:893
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.AttrPattern", AttrPatternNode, DFPatternNode)
Managed reference to AttrPatternNode.
Definition: dataflow_pattern.h:906
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(AttrPattern, DFPattern, AttrPatternNode)
AttrPattern(DFPattern pattern, DictAttrs attrs)
A pattern to match a callable node in Relax.
Definition: dataflow_pattern.h:469
bool varg_default_wildcard
Definition: dataflow_pattern.h:484
static void RegisterReflection()
Definition: dataflow_pattern.h:488
tvm::ffi::Array< DFPattern > args
Definition: dataflow_pattern.h:477
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.CallPattern", CallPatternNode, DFPatternNode)
DFPattern op
Definition: dataflow_pattern.h:476
Definition: dataflow_pattern.h:497
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(CallPattern, DFPattern, CallPatternNode)
CallPattern(DFPattern op, ffi::Array< DFPattern > args, bool varg_default_wildcard=false)
A Pattern to Match a Relax Constant.
Definition: dataflow_pattern.h:446
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.ConstantPattern", ConstantPatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:448
Managed reference to a ConstantPattern.
Definition: dataflow_pattern.h:460
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(ConstantPattern, DFPattern, ConstantPatternNode)
Additional constraints on the graph.
Definition: dataflow_pattern.h:164
TVM_FFI_DECLARE_OBJECT_INFO("relax.dpl.DFConstraint", DFConstraintNode, Object)
virtual std::tuple< PrimExpr, bool > AsPrimExpr(std::function< ffi::Optional< Var >(const DFPatternNode *)> match_state) const =0
Convert the constraint to a PrimExpr.
virtual ffi::Array< DFPattern > GetDependentPatterns() const =0
Return the patterns on which the constraint depends.
static constexpr const uint32_t _type_child_slots
Definition: dataflow_pattern.h:199
Definition: dataflow_pattern.h:203
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(DFConstraint, ObjectRef, DFConstraintNode)
Base type of all dataflow patterns.
Definition: dataflow_pattern.h:91
static constexpr const uint32_t _type_child_slots
Definition: dataflow_pattern.h:93
TVM_FFI_DECLARE_OBJECT_INFO("relax.dpl.DFPattern", DFPatternNode, Object)
Managed reference to dataflow patterns.
Definition: dataflow_pattern.h:101
CallPattern operator()(const std::vector< DFPattern > &args) const
Syntatic Sugar for creating a CallPattern.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(DFPattern, ObjectRef, DFPatternNode)
DataTypePattern HasDtype(const DataType &dtype) const
Syntatic Sugar for creating a DataTypePattern with a DataType.
SameShapeConstraint HasSameShapeAs(const DFPattern &other) const
Syntatic Sugar for creating a ShapePattern.
OrPattern operator|(const DFPattern &other) const
Syntatic Sugar for creating an OrPattern.
CallPattern operator()(Args &&... args) const
Syntatic Sugar for creating a CallPattern.
Definition: dataflow_pattern.h:970
AttrPattern HasAttr(const ffi::Map< ffi::String, Any > &attrs) const
Syntatic Sugar for creating an AttrPattern.
StructInfoPattern HasStructInfo(const StructInfo &struct_info) const
Syntatic Sugar for creating a StructInfoPattern.
ShapePattern HasShape(const ffi::Array< PrimExpr > &shape) const
Syntatic Sugar for creating a ShapePattern.
AndPattern operator&(const DFPattern &other) const
Syntatic Sugar for creating an AndPattern.
DFPattern dup() const
Syntatic Sugar for duplicating the current pattern.
NotPattern operator~() const
Syntatic Sugar for creating a NotPattern.
DataTypePattern HasDtype(const std::string &dtype) const
Syntatic Sugar for creating a DataTypePattern with a data type's name.
A pattern that asserting a root pattern has a certain data type.
Definition: dataflow_pattern.h:859
DFPattern pattern
Definition: dataflow_pattern.h:861
DataType dtype
Definition: dataflow_pattern.h:862
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.DataTypePattern", DataTypePatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:864
Managed reference to DataTypePatternNode.
Definition: dataflow_pattern.h:878
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(DataTypePattern, DFPattern, DataTypePatternNode)
DataTypePattern(DFPattern pattern, DataType dtype)
A Pattern to Match a Relax Dataflow Variable.
Definition: dataflow_pattern.h:401
static void RegisterReflection()
Definition: dataflow_pattern.h:403
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.DataflowVarPattern", DataflowVarPatternNode, VarPatternNode)
Managed reference to a DataflowVarPattern.
Definition: dataflow_pattern.h:415
DataflowVarPattern(ffi::String name_hint)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(DataflowVarPattern, DFPattern, DataflowVarPatternNode)
Pattern for Relax Expression.
Definition: dataflow_pattern.h:342
Expr expr
Definition: dataflow_pattern.h:344
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.ExprPattern", ExprPatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:346
Managed reference to an ExprPattern.
Definition: dataflow_pattern.h:357
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(ExprPattern, DFPattern, ExprPatternNode)
A pattern of external function.
Definition: dataflow_pattern.h:917
const ffi::String & global_symbol() const
The external function name.
Definition: dataflow_pattern.h:922
static void RegisterReflection()
Definition: dataflow_pattern.h:924
ffi::String global_symbol_
Definition: dataflow_pattern.h:919
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.ExternFuncPattern", ExternFuncPatternNode, DFPatternNode)
Managed reference to ExternFuncPatternNode.
Definition: dataflow_pattern.h:937
ExternFuncPattern(ffi::String global_symbol)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(ExternFuncPattern, DFPattern, ExternFuncPatternNode)
A pattern to match a Relax Function.
Definition: dataflow_pattern.h:534
static void RegisterReflection()
Definition: dataflow_pattern.h:545
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.FunctionPattern", FunctionPatternNode, DFPatternNode)
DFPattern body
Definition: dataflow_pattern.h:543
tvm::ffi::Array< DFPattern > params
Definition: dataflow_pattern.h:536
Managed reference to FunctionPatternNode.
Definition: dataflow_pattern.h:559
FunctionPattern(tvm::ffi::Array< DFPattern > params, DFPattern body)
Constructor.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(FunctionPattern, DFPattern, FunctionPatternNode)
A Pattern to Match a Relax Global Variable.
Definition: dataflow_pattern.h:426
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.GlobalVarPattern", GlobalVarPatternNode, DFPatternNode)
Managed reference to a GlobalVarPattern.
Definition: dataflow_pattern.h:436
GlobalVarPattern(ffi::String name_hint)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(GlobalVarPattern, DFPattern, GlobalVarPatternNode)
Pattern for rejecting a certain pattern.
Definition: dataflow_pattern.h:715
DFPattern reject
Definition: dataflow_pattern.h:717
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.NotPattern", NotPatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:719
Managed reference to NotPatternNode.
Definition: dataflow_pattern.h:730
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(NotPattern, DFPattern, NotPatternNode)
NotPattern(DFPattern reject)
Match a disjunction of other patterns.
Definition: dataflow_pattern.h:687
DFPattern left
Definition: dataflow_pattern.h:689
DFPattern right
Definition: dataflow_pattern.h:690
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.OrPattern", OrPatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:692
Managed reference to OrPatternNode.
Definition: dataflow_pattern.h:705
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(OrPattern, DFPattern, OrPatternNode)
OrPattern(DFPattern left, DFPattern right)
A context to manage the graph-level pattern matching.
Definition: dataflow_pattern.h:250
ExternUse
Constrainting matched graph with assertion to external uses.
Definition: dataflow_pattern.h:253
@ kMustNot
Definition: dataflow_pattern.h:255
@ kMay
Definition: dataflow_pattern.h:254
std::vector< DFConstraint > validation_constraints
Definition: dataflow_pattern.h:267
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.PatternContext", PatternContextNode, Object)
enum tvm::relax::PatternContextNode::ExternUse allow_extern_use
std::map< DFPattern, std::vector< std::pair< DFPattern, std::vector< PairCons > > > > edge_constraints
Definition: dataflow_pattern.h:260
std::vector< DFPattern > src_ordered
Definition: dataflow_pattern.h:264
Managed reference to a pattern context.
Definition: dataflow_pattern.h:275
void add_constraint(DFConstraint constraint)
Add a validation constraint.
Definition: dataflow_pattern.h:322
void ExitWithScope() const
The RAII-like exit of a constraint context scope.
static ffi::Optional< PatternContext > Current()
Get the constraint context object on the top of the stack.
PatternContextNode * operator->()
Definition: dataflow_pattern.h:286
void add_constraint(DFPattern producer, DFPattern consumer, PairCons cons)
Build an edge constraint between two patterns (producer and consumer).
Definition: dataflow_pattern.h:298
const PatternContextNode * operator->() const
Definition: dataflow_pattern.h:281
PatternContext(bool incremental=false)
PatternContext(ffi::UnsafeInit tag)
Definition: dataflow_pattern.h:277
PatternContext(ObjectPtr< Object > n)
Definition: dataflow_pattern.h:278
void EnterWithScope() const
The RAII-like entry of a constraint context scope.
A sequence of DFPatterns that the previous DFPattern is connected to the next one.
Definition: dataflow_pattern.h:212
TVM_FFI_DECLARE_OBJECT_INFO("relax.dpl.PatternSeq", PatternSeqNode, Object)
static void RegisterReflection()
Definition: dataflow_pattern.h:217
std::vector< PairCons > pair_constraints
Definition: dataflow_pattern.h:215
tvm::ffi::Array< DFPattern > patterns
Definition: dataflow_pattern.h:214
Managed reference to pattern sequences.
Definition: dataflow_pattern.h:228
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(PatternSeq, ObjectRef, PatternSeqNode)
PatternSeq OnlyUsedBy(PatternSeq other, int index=-1) const
PatternSeq UsedBy(PatternSeq other, int index=-1) const
PatternSeq(tvm::ffi::Array< DFPattern > patterns, bool only_used_by=false)
friend PatternSeq UsedBy(const PatternSeq &lhs, const PatternSeq &rhs, int index)
Create used-by relationship between lhs[-1] and rhs[0], with [*lhs, *rhs] returned.
PatternSeq(DFPattern init_pattern)
friend PatternSeq OnlyUsedBy(const PatternSeq &lhs, const PatternSeq &rhs, int index)
Create only-used-by relationship between lhs[-1] and rhs[0], with [*lhs, *rhs] returned.
PatternSeq dup() const
Syntatic Sugar for duplicating the current pattern sequence.
A pattern to match an array of PrimExpr.
Definition: dataflow_pattern.h:508
ffi::Array< PrimExpr > fields
Definition: dataflow_pattern.h:510
static void RegisterReflection()
Definition: dataflow_pattern.h:512
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.PrimArrPattern", PrimArrPatternNode, DFPatternNode)
Managed reference to a PrimArrPattern.
Definition: dataflow_pattern.h:523
PrimArrPattern(ffi::Array< PrimExpr > arr)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(PrimArrPattern, DFPattern, PrimArrPatternNode)
A pattern that asserting multiple root patterns have the same shape.
Definition: dataflow_pattern.h:827
ffi::Array< DFPattern > GetDependentPatterns() const override
Return the patterns on which the constraint depends.
Definition: dataflow_pattern.h:831
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.SameShapeConstraint", SameShapeConstraintNode, DFConstraintNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:836
std::tuple< PrimExpr, bool > AsPrimExpr(std::function< ffi::Optional< Var >(const DFPatternNode *)> match_state) const override
Convert the constraint to a PrimExpr.
ffi::Array< DFPattern > args
Definition: dataflow_pattern.h:829
Managed reference to SameShapePatternNode.
Definition: dataflow_pattern.h:848
SameShapeConstraint(ffi::Array< DFPattern > args)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(SameShapeConstraint, DFConstraint, SameShapeConstraintNode)
A pattern that asserting a root pattern has a certain shape.
Definition: dataflow_pattern.h:799
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.ShapePattern", ShapePatternNode, DFPatternNode)
ffi::Array< PrimExpr > shape
Definition: dataflow_pattern.h:802
DFPattern pattern
Definition: dataflow_pattern.h:801
static void RegisterReflection()
Definition: dataflow_pattern.h:804
Managed reference to ShapePatternNode.
Definition: dataflow_pattern.h:817
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(ShapePattern, DFPattern, ShapePatternNode)
ShapePattern(DFPattern pattern, ffi::Array< PrimExpr > type)
Pattern for matching a certain struct info.
Definition: dataflow_pattern.h:774
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.StructInfoPattern", StructInfoPatternNode, DFPatternNode)
StructInfo struct_info
Definition: dataflow_pattern.h:777
static void RegisterReflection()
Definition: dataflow_pattern.h:779
DFPattern pattern
Definition: dataflow_pattern.h:776
Definition: dataflow_pattern.h:789
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(StructInfoPattern, DFPattern, StructInfoPatternNode)
StructInfoPattern(DFPattern pattern, StructInfo struct_info)
Managed reference to StructInfoNode.
Definition: expr.h:132
A pattern to match n'th indexing to a tuple.
Definition: dataflow_pattern.h:629
static void RegisterReflection()
Definition: dataflow_pattern.h:634
DFPattern tuple
Definition: dataflow_pattern.h:631
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.TupleGetItemPattern", TupleGetItemPatternNode, DFPatternNode)
int index
Definition: dataflow_pattern.h:632
Managed reference to TupleGetItemPatternNode.
Definition: dataflow_pattern.h:648
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(TupleGetItemPattern, DFPattern, TupleGetItemPatternNode)
TupleGetItemPattern(DFPattern tuple, int index)
Pattern to match a tuple of ordered expressions.
Definition: dataflow_pattern.h:575
tvm::ffi::Array< DFPattern > fields
Definition: dataflow_pattern.h:577
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.TuplePattern", TuplePatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:579
Managed reference to TuplePatternNode.
Definition: dataflow_pattern.h:590
TuplePattern(tvm::ffi::Array< DFPattern > fields)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(TuplePattern, DFPattern, TuplePatternNode)
A pattern to match multiple expressions unorderedly.
Definition: dataflow_pattern.h:600
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.UnorderedTuplePattern", UnorderedTuplePatternNode, DFPatternNode)
tvm::ffi::Array< DFPattern > fields
Definition: dataflow_pattern.h:602
static void RegisterReflection()
Definition: dataflow_pattern.h:604
Managed reference to UnorderedTuplePatternNode.
Definition: dataflow_pattern.h:617
UnorderedTuplePattern(tvm::ffi::Array< DFPattern > fields)
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(UnorderedTuplePattern, DFPattern, UnorderedTuplePatternNode)
A Pattern to Match a Relax Variable.
Definition: dataflow_pattern.h:368
static constexpr const uint32_t _type_child_slots
Definition: dataflow_pattern.h:378
ffi::String name
Definition: dataflow_pattern.h:370
const ffi::String & name_hint() const
Definition: dataflow_pattern.h:371
TVM_FFI_DECLARE_OBJECT_INFO("relax.dpl.VarPattern", VarPatternNode, DFPatternNode)
static void RegisterReflection()
Definition: dataflow_pattern.h:373
Managed reference to a VarPattern.
Definition: dataflow_pattern.h:386
VarPattern(ffi::String name_hint)
Create a pattern matching by variable name.
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE(VarPattern, DFPattern, VarPatternNode)
Wildcard Pattern is a pattern that can match anything.
Definition: dataflow_pattern.h:740
static void RegisterReflection()
Definition: dataflow_pattern.h:742
TVM_FFI_DECLARE_OBJECT_INFO_FINAL("relax.dpl.WildcardPattern", WildcardPatternNode, DFPatternNode)
Managed reference to WildcardPatternNode.
Definition: dataflow_pattern.h:754
TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE(WildcardPattern, DFPattern, WildcardPatternNode)
WildcardPattern(ObjectPtr< WildcardPatternNode > data)
Definition: dataflow_pattern.h:757
Runtime primitive data type.
Definition: data_type.h:47
Definition: repr_printer.h:91
PatternSeq operator^(const PatternSeq &lhs, const PatternSeq &rhs)
Syntax sugar of UsedBy(lhs, rhs, -1).
ExprPattern IsOp(const ffi::String &op_name)
Syntatic Sugar for creating a ExprPattern base on an Op.
DFPattern IsTuple(const ffi::Array< DFPattern > &fields, bool unordered=false)
Syntatic Sugar for creating TuplePattern or UnorderedTuplePattern (unordered=true)
CallPattern IsCallTIR(const ffi::String &name, ffi::Optional< TuplePattern > args=std::nullopt)
Syntatic Sugar for call_tir (return a tensor)
CallPattern IsCallDPSPacked(const ffi::String &name, ffi::Optional< TuplePattern > args=std::nullopt)
Syntatic Sugar for call_dps_packed (return a tensor)
WildcardPattern Wildcard()
Syntatic Sugar for creating a WildcardPattern.
VarPattern IsVar(const ffi::String &name)
Syntatic Sugar for creating a VarPattern with a name.
ConstantPattern IsConst()
Syntatic Sugar for creating a ConstantPattern.
PatternSeq UsedBy(const PatternSeq &lhs, const PatternSeq &rhs, int index=-1)
Create used-by relationship between lhs[-1] and rhs[0], with [*lhs, *rhs] returned.
PatternSeq OnlyUsedBy(const PatternSeq &lhs, const PatternSeq &rhs, int index=-1)
Create only-used-by relationship between lhs[-1] and rhs[0], with [*lhs, *rhs] returned.
ExprPattern IsExpr(const Expr &expr)
Syntatic Sugar for creating a ExprPattern.
PatternSeq operator>>(const PatternSeq &lhs, const PatternSeq &rhs)
Syntax sugar of OnlyUsedBy(lhs, rhs, -1).
TupleGetItemPattern IsTupleGetItem(const DFPattern tuple, int index=-1)
Syntatic Sugar for creating a TupleGetItemPattern.
Tensor shape(const Tensor &src, DataType dtype, const std::string name="T_shape", const std::string tag=kInjective)
Get the shape of input tensor.
Definition: transform.h:1960
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37
Constraint of a DFPattern edge (producer -> consumer) in graph-level matching.
Definition: dataflow_pattern.h:136
int index
Definition: dataflow_pattern.h:142
bool operator==(const PairCons &other) const
Definition: dataflow_pattern.h:152
enum tvm::relax::PairCons::Type type
Type
Constraint types of the edge.
Definition: dataflow_pattern.h:138
@ kOnlyUsedBy
Definition: dataflow_pattern.h:140
@ kUsedBy
Definition: dataflow_pattern.h:139
PairCons(Type t, int index=-1)
Construct a new PairCons object.
Definition: dataflow_pattern.h:150
RAII wrapper function to enter and exit a context object similar to python's with syntax.