Home
last modified time | relevance | path

Searched refs:axis_read (Results 1 – 1 of 1) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py501 def axis_read(axis_env, axis_name): function
510 mesh_axes = tuple(unsafe_map(partial(axis_read, axis_env), name))