| 3385 | XlaOp ReplicaId(XlaBuilder* builder) { return builder->ReplicaId(); } |
| 3386 | |
| 3387 | XlaOp SelectAndScatter(const XlaOp operand, const XlaComputation& select, |
| 3388 | absl::Span<const int64> window_dimensions, |
| 3389 | absl::Span<const int64> window_strides, Padding padding, |
| 3390 | const XlaOp source, const XlaOp init_value, |
| 3391 | const XlaComputation& scatter) { |
| 3392 | return operand.builder()->SelectAndScatter(operand, select, window_dimensions, |
| 3393 | window_strides, padding, source, |
| 3394 | init_value, scatter); |
| 3395 | } |
| 3396 | |
| 3397 | XlaOp SelectAndScatterWithGeneralPadding( |
| 3398 | const XlaOp operand, const XlaComputation& select, |