| 3313 | } |
| 3314 | |
| 3315 | XlaOp Reduce(const XlaOp operand, const XlaOp init_value, |
| 3316 | const XlaComputation& computation, |
| 3317 | absl::Span<const int64> dimensions_to_reduce) { |
| 3318 | return operand.builder()->Reduce(operand, init_value, computation, |
| 3319 | dimensions_to_reduce); |
| 3320 | } |
| 3321 | |
| 3322 | // Reduces several arrays simultaneously among the provided dimensions, given |
| 3323 | // "computation" as a reduction operator. |