| 144 | } |
| 145 | |
| 146 | void BuildReducer(xla::XlaBuilder* builder, const xla::XlaOp& scalar_lhs, |
| 147 | const xla::XlaOp& scalar_rhs) override { |
| 148 | xla::And(scalar_lhs, scalar_rhs); |
| 149 | } |
| 150 | }; |
| 151 | |
| 152 | REGISTER_XLA_OP(Name("All").CompileTimeConstantInput("reduction_indices"), |