| 98 | } |
| 99 | |
| 100 | XlaOp Any(XlaOp predicates) { |
| 101 | XlaBuilder* builder = predicates.builder(); |
| 102 | return builder->ReportErrorOrReturn([&]() -> StatusOr<XlaOp> { |
| 103 | auto f = ConstantR0<bool>(builder, false); |
| 104 | XlaComputation logical_or = CreateScalarOrComputation(PRED, builder); |
| 105 | TF_ASSIGN_OR_RETURN(const Shape& predicates_shape, |
| 106 | builder->GetShape(predicates)); |
| 107 | std::vector<int64> all_dimensions(predicates_shape.rank()); |
| 108 | std::iota(all_dimensions.begin(), all_dimensions.end(), 0); |
| 109 | return Reduce(predicates, f, logical_or, all_dimensions); |
| 110 | }); |
| 111 | } |
| 112 | |
| 113 | namespace { |
| 114 | |