Ctrl+K

jax.lax.reduce_precision

jax.lax.reduce_precision#

jax.lax.reduce_precision(operand, exponent_bits, mantissa_bits)[source]#

Wraps XLA’s ReducePrecision operator.

Parameters:

  • operand (float | ArrayLike) –

  • exponent_bits (int) –

  • mantissa_bits (int) –

Return type:

Array

Contents