Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dxla.py469 out_nodes = xla_destructure(c, ans)
477 def xla_destructure(c, ans): function
821 xla_inputs = iter(xla_destructure(c, tuple_param))
/dports/math/py-jax/jax-0.2.9/jax/_src/lax/
H A Dparallel.py1038 out, = xla.xla_destructure(c, out_tup)