MCPcopy Create free account
hub / github.com/apache/tvm / __init__

Method __init__

python/tvm/tirx/predicate.py:38–41  ·  view source on GitHub ↗
(self, f_pred: Callable[..., PrimExpr])

Source from the content-addressed store, hash-verified

36 pred: PrimExpr
37
38 def __init__(self, f_pred: Callable[..., PrimExpr]):
39 vars = [Var(name, "int32") for name in inspect.signature(f_pred).parameters]
40 pred = f_pred(*vars)
41 self.__init_handle_by_constructor__(_ffi_api.Predicate, vars, pred)
42
43 def apply(self, indices: list[PrimExpr]) -> PrimExpr:
44 """Apply the predicate to the given indices"""

Callers

nothing calls this directly

Calls 1

VarClass · 0.90

Tested by

no test coverage detected