View source on GitHub
|
Evaluates expressions into JAX values.
oryx.experimental.matching.jax_rewrite.evaluate(
obj,
env: oryx.experimental.matching.jax_rewrite.Bindings
) -> Any
evaluate is a single-dispatch function whose default implementation is to
raise a NotImplementedError. To register evaluation functions for particular
types, use evaluate.register.
Args | |
|---|---|
obj
|
an object to evaluate. |
env
|
a dictionary mapping string names to JAX values |
Returns | |
|---|---|
| The JAX value that is the result of evaluating the object. |