jax.lax.reduce_precision

Warning

This page was created from a pull request (#9655).

jax.lax.reduce_precision¶

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

Wraps XLA’s ReducePrecision operator.

Parameters

  • operand (Union[float, Any]) –

  • exponent_bits (int) –

  • mantissa_bits (int) –

Return type

Any

previous

jax.lax.reduce

next

jax.lax.reduce_window