Skip to main content

KeepAsOp

Type Alias KeepAsOp 

Source
pub type KeepAsOp<'a> = dyn Fn(&Node, u64, &[DataType]) -> bool + 'a;
Expand description

Predicate deciding whether a matched function-call node should be kept as an op (a fused kernel claims it) instead of being inlined into its body.

Receives a lightweight claim-view Node (op type, normalized domain, and scalar attributes — enough for an EP’s supports_op claim gate), the effective opset for the node’s domain, and the node’s positional input dtypes (resolved from the model’s value_info/inputs/initializers; DataType::Undefined where unknown). Returning true keeps the node; false (or an unresolved dtype the caller declines on) inlines it, so the default behavior is preserved for every op no kernel claims.