| 91 | } |
| 92 | |
| 93 | void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, |
| 94 | const xla::XlaOp& scalar_rhs) override { |
| 95 | xla::Max(scalar_lhs, scalar_rhs); |
| 96 | } |
| 97 | }; |
| 98 | |
| 99 | REGISTER_XLA_OP(Name("Max").CompileTimeConstantInput("reduction_indices"), |