Home
last modified time | relevance | path

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

/dports/math/py-jax/jax-0.2.9/jax/
H A Dapi.py2452 out_flat = pe.remat_call(flat_fun, *args_flat, name=flat_fun.__name__,
/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dpartial_eval.py748 remat_call = remat_call_p.bind variable