Skip to content

Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's #70418

Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's

Add a sharding rule for reduce_precision_p and properly thread eqn.ctx in loops.py where we create pe.new_jaxpr_eqn's #70418

Annotations

1 warning

jax2tf_test (py 3.10 on ubuntu-latest, x64=0)

succeeded Jan 18, 2025 in 1m 14s