tvm
|
#include <dataflow_pattern.h>
Public Member Functions | |
CallPattern (DFPattern op, Array< DFPattern > args, bool varg_default_wildcard=false) | |
TVM_DEFINE_OBJECT_REF_METHODS (CallPattern, DFPattern, CallPatternNode) | |
![]() | |
template<typename... Args> | |
CallPattern | operator() (Args &&... args) const |
Syntatic Sugar for creating a CallPattern. More... | |
CallPattern | operator() (const std::vector< DFPattern > &args) const |
Syntatic Sugar for creating a CallPattern. More... | |
OrPattern | operator| (const DFPattern &other) const |
Syntatic Sugar for creating an OrPattern. More... | |
AndPattern | operator& (const DFPattern &other) const |
Syntatic Sugar for creating an AndPattern. More... | |
NotPattern | operator~ () const |
Syntatic Sugar for creating a NotPattern. More... | |
AttrPattern | HasAttr (const Map< String, Any > &attrs) const |
Syntatic Sugar for creating an AttrPattern. More... | |
StructInfoPattern | HasStructInfo (const StructInfo &struct_info) const |
Syntatic Sugar for creating a StructInfoPattern. More... | |
DataTypePattern | HasDtype (const DataType &dtype) const |
Syntatic Sugar for creating a DataTypePattern with a DataType. More... | |
DataTypePattern | HasDtype (const std::string &dtype) const |
Syntatic Sugar for creating a DataTypePattern with a data type's name. More... | |
ShapePattern | HasShape (const Array< PrimExpr > &shape) const |
Syntatic Sugar for creating a ShapePattern. More... | |
SameShapeConstraint | HasSameShapeAs (const DFPattern &other) const |
Syntatic Sugar for creating a ShapePattern. More... | |
DFPattern | dup () const |
Syntatic Sugar for duplicating the current pattern. More... | |
operator PatternSeq () const | |
Implicit conversion from DFPattern to PatternSeq. More... | |
TVM_DEFINE_OBJECT_REF_METHODS (DFPattern, ObjectRef, DFPatternNode) | |
tvm::relax::CallPattern::CallPattern | ( | DFPattern | op, |
Array< DFPattern > | args, | ||
bool | varg_default_wildcard = false |
||
) |
tvm::relax::CallPattern::TVM_DEFINE_OBJECT_REF_METHODS | ( | CallPattern | , |
DFPattern | , | ||
CallPatternNode | |||
) |