View source on GitHub |
An expression that looks up a provided name in the environment.
Inherits From: JaxExpression
, Expression
, Pattern
oryx.experimental.matching.jax_rewrite.JaxVar(
name: str, shape: Tuple[Any], dtype: jnp.dtype
)
Attributes | |
---|---|
name
|
The string name for the JaxVar used to look up the JaxVar 's value
in an environment.
|
dtype
|
|
shape
|
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(
_
) -> 'JaxVar'