Dedup prev_outs with cur_inps so we don't barrier on the same input twice. In other words, if some prev_outs are in cur_inps, then skip those prev_outs from opt_barrier (since they are already covered by cur_inps).
PiperOrigin-RevId: 972657558
Y
Yash Katariya committed
ae115c22043f2f4b77defdaa688aae7d072a440b
Parent: 1214954
Committed by jax authors <google-ml-automation@google.com>
on 8/28/2026, 5:18:42 PM