tvm_tirx.tirx.structural.structural_visit_with_constraint_context

tvm_tirx.tirx.structural.structural_visit_with_constraint_context(root, callback)

Visit a value structurally with a callback that owns child traversal.

The callback receives (value, visitor). It may call visitor.visit for selected children or visitor.default_visit for the value’s default descent. Returning without either call prunes that value’s subtree. visitor.constraints contains the active symbolic facts.

Parameters:

• root (Any) – Root value to traverse.

• callback (callable) – Called as callback(value, visitor) with a StructuralVisitorWithConstraintContext. Return None to continue or a tvm_ffi.VisitInterrupt to stop; propagate child interrupts.

Returns:

result (tvm_ffi.VisitInterrupt or None) – The interrupt returned by a callback, or None if traversal completes.