A pattern that asserting multiple root patterns have the same shape.
More...
#include <dataflow_pattern.h>
|
static constexpr const char * | _type_key = "relax.dpl.SameShapeConstraint" |
|
static constexpr const char * | _type_key = "DFConstraintNode" |
|
static constexpr const uint32_t | _type_child_slots = 1 |
|
A pattern that asserting multiple root patterns have the same shape.
- See also
- SameShapePattern
◆ AsPrimExpr()
std::tuple<PrimExpr, bool> tvm::relax::SameShapeConstraintNode::AsPrimExpr |
( |
std::function< Optional< Var >(const DFPatternNode *)> |
match_state | ) |
const |
|
overridevirtual |
Convert the constraint to a PrimExpr.
If the returned boolean parameter is true, then the returned expression is a necessary-and-sufficient condition for evaluating the constraint. In this case, the matcher may either mark the constraint as satisfied (no need to re-check later), or as failed (need to back-track).
If the returned boolean parameter is false, then the returned expression is a necessary-but-not-sufficient condition for evaluating the constraint. In this case, the matcher may start backtracking as a result of a failed condition, but may not mark the constraint as satisfied. This typically occurs when the constraint involves a parameter that the matcher has not yet filled.
- Parameters
-
match_state | A function that can be called to check the current state of the match. The function takes as argument a pattern on which the constraint depends, and returns the relax variable matched by that pattern, or std::nullopt if the pattern has not yet been matched. |
- Returns
- A tuple of
PrimExpr
and bool
. The first element is a necessary condition for the constraint to be satisfied. The second tuple element indicates whether the condition is also sufficient for the constraint to be satisfied.
Implements tvm::relax::DFConstraintNode.
◆ GetDependentPatterns()
Array<DFPattern> tvm::relax::SameShapeConstraintNode::GetDependentPatterns |
( |
| ) |
const |
|
inlineoverridevirtual |
◆ RegisterReflection()
static void tvm::relax::SameShapeConstraintNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
◆ _type_key
constexpr const char* tvm::relax::SameShapeConstraintNode::_type_key = "relax.dpl.SameShapeConstraint" |
|
staticconstexpr |
◆ args
Array<DFPattern> tvm::relax::SameShapeConstraintNode::args |
The patterns with matching shapes
The documentation for this class was generated from the following file: