tvm
|
Managed reference to dataflow patterns. More...
#include <dataflow_pattern.h>
Public Member Functions | |
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) | |
Managed reference to dataflow patterns.
DFPattern tvm::relax::DFPattern::dup | ( | ) | const |
Syntatic Sugar for duplicating the current pattern.
AttrPattern tvm::relax::DFPattern::HasAttr | ( | const Map< String, Any > & | attrs | ) | const |
Syntatic Sugar for creating an AttrPattern.
DataTypePattern tvm::relax::DFPattern::HasDtype | ( | const DataType & | dtype | ) | const |
Syntatic Sugar for creating a DataTypePattern with a DataType.
DataTypePattern tvm::relax::DFPattern::HasDtype | ( | const std::string & | dtype | ) | const |
Syntatic Sugar for creating a DataTypePattern with a data type's name.
SameShapeConstraint tvm::relax::DFPattern::HasSameShapeAs | ( | const DFPattern & | other | ) | const |
Syntatic Sugar for creating a ShapePattern.
ShapePattern tvm::relax::DFPattern::HasShape | ( | const Array< PrimExpr > & | shape | ) | const |
Syntatic Sugar for creating a ShapePattern.
StructInfoPattern tvm::relax::DFPattern::HasStructInfo | ( | const StructInfo & | struct_info | ) | const |
Syntatic Sugar for creating a StructInfoPattern.
tvm::relax::DFPattern::operator PatternSeq | ( | ) | const |
Implicit conversion from DFPattern to PatternSeq.
AndPattern tvm::relax::DFPattern::operator& | ( | const DFPattern & | other | ) | const |
Syntatic Sugar for creating an AndPattern.
CallPattern tvm::relax::DFPattern::operator() | ( | Args &&... | args | ) | const |
Syntatic Sugar for creating a CallPattern.
Implementation of the templated CallPattern syntax sugar.
CallPattern tvm::relax::DFPattern::operator() | ( | const std::vector< DFPattern > & | args | ) | const |
Syntatic Sugar for creating a CallPattern.
Syntatic Sugar for creating an OrPattern.
NotPattern tvm::relax::DFPattern::operator~ | ( | ) | const |
Syntatic Sugar for creating a NotPattern.
tvm::relax::DFPattern::TVM_DEFINE_OBJECT_REF_METHODS | ( | DFPattern | , |
ObjectRef | , | ||
DFPatternNode | |||
) |