Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py81 def _make_unit_shape(_): return (xc.Shape.array_shape(np.dtype('bool'), ()),) function
103 core.AbstractUnit: _make_unit_shape,