jax
b19a213c - Reland _convert_element_type fast path with two changes:

Commit
12 days ago
Reland _convert_element_type fast path with two changes: * remove the ndarray case, which seems to be causing downstream problems. * tolerate OverflowError exceptions and skip the fast path if seen. Reverts 78316cdc494bdfecf1e3adc454119bd6f8445d30 PiperOrigin-RevId: 905164369
Author
Parents
Loading