Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py372 _check_special(prim.name, buf.xla_shape(), buf)
374 def _check_special(name, xla_shape, buf): function