ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

oryx.experimental.matching.jax_rewrite.JaxVar

An expression that looks up a provided name in the environment.

Inherits From: JaxExpression, Expression, Pattern

name a string name corresponding the key used to lookup this JaxVar's value in an environment.
shape a tuple of integers corresponding to the shape of the JaxVar.
dtype a JAX dtype object for this JaxVar's dtype.

name The string name for the JaxVar used to look up the JaxVar's value in an environment.
dtype

shape

Methods

evaluate

View source

match

View source

tree_children

View source

tree_map

View source