Searched refs:_xla_consts (Results 1 – 1 of 1) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 642 def _xla_consts(c, consts): function
706 xla_consts = _xla_consts(c, consts)
957 outs = jaxpr_subcomp(c, jaxpr, None, axis_env, _xla_consts(c, consts), '',
963 xla_consts = _xla_consts(c, consts)
984 outs = jaxpr_subcomp(c, jaxpr, backend, axis_env, _xla_consts(c, consts),
990 xla_consts = _xla_consts(c, consts)
|