Home
last modified time | relevance | path

Searched defs:JaxprTracer (Results 1 – 1 of 1) sorted by relevance

/dports/math/py-jax/jax-0.2.9/jax/interpreters/
H A Dpartial_eval.py420 class JaxprTracer(Tracer): class
546 def new_eqn_recipe(invars: Sequence[JaxprTracer],
547 outvars: Sequence[JaxprTracer],
572 def recipe_to_eqn(getvar: Callable[[JaxprTracer], core.Atom],
582 in_tracers: Sequence[JaxprTracer],
583 out_tracers: Sequence[JaxprTracer]
598 def getvar(t: JaxprTracer) -> core.Atom: