Searched defs:trace_to_jaxpr (Results 1 – 1 of 1) sorted by relevance
459 def trace_to_jaxpr(fun: lu.WrappedFun, pvals: Sequence[PartialVal], function1245 def trace_to_jaxpr(fun: lu.WrappedFun, pvals: Sequence[PartialVal], function