View source on GitHub
|
A node in an expression tree.
Inherits From: Expression, Pattern
JAXExpressions are subclasses of rules.Expression so if the tree_map
and tree_children methods are implemented, it is compatible with the
rewrite interface in rules. To be compatible with the pattern matching
interface in matcher, the match method needs to be implemented too.
Attributes | |
|---|---|
dtype
|
|
shape
|
|
Methods
evaluate
@abc.abstractmethodevaluate( 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()
tree_map
tree_map(
fn
)
View source on GitHub