tvm
|
A pattern which matches attributes in another pattern. More...
#include <dataflow_pattern.h>
Public Member Functions | |
AttrPattern (DFPattern pattern, DictAttrs attrs) | |
TVM_DEFINE_OBJECT_REF_METHODS (AttrPattern, DFPattern, AttrPatternNode) | |
Public Member Functions inherited from tvm::relay::DFPattern | |
DFPattern | operator() (const std::vector< DFPattern > &args) const |
Syntatic Sugar for creating a CallPattern. More... | |
DFPattern | operator+ (const DFPattern &other) const |
Syntatic Sugar for creating a CallPattern with an "add" op. More... | |
DFPattern | operator- (const DFPattern &other) const |
Syntatic Sugar for creating a CallPattern with a "subtract" op. More... | |
DFPattern | operator* (const DFPattern &other) const |
Syntatic Sugar for creating a CallPattern with a "multiply" op. More... | |
DFPattern | operator/ (const DFPattern &other) const |
Syntatic Sugar for creating a CallPattern with a "divide" op. More... | |
DFPattern | operator|| (const DFPattern &other) const |
Syntatic Sugar for creating an AltPattern. More... | |
DFPattern | Optional (const std::function< DFPattern(const DFPattern &)> &func) const |
Syntatic Sugar for creating an Optional Pattern. More... | |
DFPattern | HasAttr (const Map< String, ObjectRef > &attrs) const |
Syntatic Sugar for creating an AttrPattern. More... | |
DFPattern | HasType (const Type &type) const |
Syntatic Sugar for creating a TypePattern. More... | |
DFPattern | HasDtype (const DataType &dtype) const |
Syntatic Sugar for creating a DataTypePattern with a DataType. More... | |
DFPattern | HasDtype (const std::string &dtype) const |
Syntatic Sugar for creating a DataTypePattern with a data type's name. More... | |
DFPattern | HasShape (const Array< PrimExpr > shape) const |
Syntatic Sugar for creating a ShapePattern. More... | |
TVM_DEFINE_OBJECT_REF_METHODS (DFPattern, ObjectRef, DFPatternNode) | |
Public Member Functions inherited from tvm::runtime::ObjectRef | |
ObjectRef ()=default | |
default constructor More... | |
ObjectRef (ObjectPtr< Object > data) | |
Constructor from existing object ptr. More... | |
bool | same_as (const ObjectRef &other) const |
Comparator. More... | |
bool | operator== (const ObjectRef &other) const |
Comparator. More... | |
bool | operator!= (const ObjectRef &other) const |
Comparator. More... | |
bool | operator< (const ObjectRef &other) const |
Comparator. More... | |
bool | defined () const |
const Object * | get () const |
const Object * | operator-> () const |
bool | unique () const |
int | use_count () const |
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>> | |
const ObjectType * | as () const |
Try to downcast the internal Object to a raw pointer of a corresponding type. More... | |
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>> | |
Optional< ObjectRefType > | as () const |
Try to downcast the ObjectRef to a Optional<T> of the requested type. More... | |
Additional Inherited Members | |
Public Types inherited from tvm::runtime::ObjectRef | |
using | ContainerType = Object |
type indicate the container type. More... | |
Static Public Attributes inherited from tvm::runtime::ObjectRef | |
static constexpr bool | _type_is_nullable = true |
Protected Member Functions inherited from tvm::runtime::ObjectRef | |
Object * | get_mutable () const |
Static Protected Member Functions inherited from tvm::runtime::ObjectRef | |
template<typename T > | |
static T | DowncastNoCheck (ObjectRef ref) |
Internal helper function downcast a ref without check. More... | |
static void | FFIClearAfterMove (ObjectRef *ref) |
Clear the object ref data field without DecRef after we successfully moved the field. More... | |
template<typename ObjectType > | |
static ObjectPtr< ObjectType > | GetDataPtr (const ObjectRef &ref) |
Internal helper function get data_ as ObjectPtr of ObjectType. More... | |
Protected Attributes inherited from tvm::runtime::ObjectRef | |
ObjectPtr< Object > | data_ |
Internal pointer that backs the reference. More... | |
A pattern which matches attributes in another pattern.
tvm::relay::AttrPattern::TVM_DEFINE_OBJECT_REF_METHODS | ( | AttrPattern | , |
DFPattern | , | ||
AttrPatternNode | |||
) |