| 73 | } |
| 74 | |
| 75 | void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, |
| 76 | const xla::XlaOp& scalar_rhs) override { |
| 77 | xla::Min(scalar_lhs, scalar_rhs); |
| 78 | } |
| 79 | }; |
| 80 | |
| 81 | REGISTER_XLA_OP(Name("Min").CompileTimeConstantInput("reduction_indices"), |