jax
jax copied to clipboard
[mhlo] Unify different versions of round op (viz., mhlo::round_nearest_afz & mhlo::round_nearest_even) into one.
[mhlo] Unify different versions of round op (viz., mhlo::round_nearest_afz & mhlo::round_nearest_even) into one.
Currently in MHLO, we have two version of round ops with different rounding modes. This CL unifies them into one op "mhlo::round" with an additional attribute to specify the rounding mode.