Searched defs:in_tracers (Results 1 – 2 of 2) sorted by relevance
457 def trace_to_jaxpr_finalize(in_tracers, out_tracers, trace, instantiate=True): argument
954 def to_jaxpr(self, in_tracers, out_tracers): argument