tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::relax::CallPatternNode Class Reference

A pattern to match a callable node in Relax. More...

#include <dataflow_pattern.h>

Inheritance diagram for tvm::relax::CallPatternNode:
Collaboration diagram for tvm::relax::CallPatternNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (CallPatternNode, DFPatternNode)
 
- Public Member Functions inherited from tvm::relax::DFPatternNode
 TVM_DECLARE_BASE_OBJECT_INFO (DFPatternNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

DFPattern op
 
tvm::Array< DFPatternargs
 
bool varg_default_wildcard
 

Static Public Attributes

static constexpr const char * _type_key = "relax.dpl.CallPattern"
 
- Static Public Attributes inherited from tvm::relax::DFPatternNode
static constexpr const char * _type_key = "DFPatternNode"
 
static constexpr const uint32_t _type_child_slots = 21
 

Detailed Description

A pattern to match a callable node in Relax.

See also
CallPattern

Member Function Documentation

◆ RegisterReflection()

static void tvm::relax::CallPatternNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::CallPatternNode::TVM_DECLARE_FINAL_OBJECT_INFO ( CallPatternNode  ,
DFPatternNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::relax::CallPatternNode::_type_key = "relax.dpl.CallPattern"
staticconstexpr

◆ args

tvm::Array<DFPattern> tvm::relax::CallPatternNode::args

The arguments of the function call

◆ op

DFPattern tvm::relax::CallPatternNode::op
Note
The op field can be:
  • relax::Op which corresponds to the primitive operators.
  • user defined functions (Function, GlobalVar, Var). The operator (function) being invoked

◆ varg_default_wildcard

bool tvm::relax::CallPatternNode::varg_default_wildcard
Note
If varg_default_wildcard is true. Given args of [pA, pB], when matching a call whose arguments are [A, B, ...], the pattern will still match despite N(args) < N(call.args). That said, with varg_default_wildcard set to true, we match the args in the order we have, and regard the rest of the arguments as wildcards. N(args) can be < N(real args) by the padding of Wildcard

The documentation for this class was generated from the following file: