Skip to main content
Ctrl+K

jax.lax.reduce_precision#

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

Wraps XLA’s ReducePrecision operator.

Parameters:
  • operand (Union[float, Array, ndarray, bool_, number, bool, int, complex]) –

  • exponent_bits (int) –

  • mantissa_bits (int) –

Return type:

Array

previous

jax.lax.reduce

next

jax.lax.reduce_window