jax.lax.reduce_precision

jax.lax.reduce_precision(operand, exponent_bits, mantissa_bits)

Wraps XLA’s ReducePrecision operator.

Parameters:

• operand (float | ArrayLike)

• exponent_bits (int)

• mantissa_bits (int)

Return type:

Array