Searched defs:partial_eval (Results 1 – 13 of 13) sorted by relevance
1552 partial_eval(Expr) -> function
1542 partial_eval(Expr) -> function
1568 partial_eval(Expr) -> function
293 def partial_eval(self, f: lu.WrappedFun, pvals: Sequence[PartialVal], member in JaxprTrace
105 namespace partial_eval { namespace
103 namespace partial_eval { namespace
283 partial_eval(acc_cond::mark_t infinitely_often, in partial_eval() function