Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py514 return _axis_groups(mesh_spec, mesh_axes)
516 def _axis_groups(mesh_spec, mesh_axes): function