Searched refs:eqn_replicas (Results 1 – 1 of 1) sorted by relevance
538 return max(unsafe_map(eqn_replicas, jaxpr.eqns), default=1)542 def eqn_replicas(eqn): function