View source on GitHub |
A JAXExpression
corresponding to a jax.core.Primitive
.
Inherits From: JaxExpression
, Expression
, Pattern
oryx.experimental.matching.jax_rewrite.Primitive(
primitive: Union[oryx.experimental.matching.matcher.Pattern
, jax_core.Primitive],
operands: Tuple[Any, ...],
params: Union[oryx.experimental.matching.matcher.Pattern
, oryx.experimental.matching.jax_rewrite.Params
]
)
A Primitive
encapsulates a jax.core.Primitive
, its arguments and its
parameters.
Methods
evaluate
evaluate(
env: oryx.experimental.matching.jax_rewrite.Bindings
) -> Any
match
match(
expr: Expr,
bindings: oryx.experimental.matching.jax_rewrite.Bindings
,
succeed: oryx.experimental.matching.jax_rewrite.Continuation
) -> oryx.experimental.matching.jax_rewrite.Success
tree_children
tree_children() -> Iterator[Expr]
tree_map
tree_map(
fn
) -> 'Primitive'
__eq__
__eq__(
other: Any
) -> bool
Return self==value.