| 781 | } |
| 782 | |
| 783 | Status ShapeVerifier::HandleReduceWindow(HloInstruction* reduce_window) { |
| 784 | TF_RETURN_IF_ERROR(CheckShape( |
| 785 | reduce_window, |
| 786 | ShapeInference::InferReduceWindowShape( |
| 787 | reduce_window->operand(0)->shape(), |
| 788 | reduce_window->operand(1)->shape(), reduce_window->window(), |
| 789 | reduce_window->to_apply()->ComputeProgramShape()))); |
| 790 | |
| 791 | return allow_mixed_precision_ |
| 792 | ? Status::OK() |
| 793 | : SameElementTypesForOperandsAndToApplyParameters(*reduce_window, |
| 794 | 1); |
| 795 | } |
| 796 | |
| 797 | Status ShapeVerifier::HandleSelectAndScatter(HloInstruction* instruction) { |
| 798 | return CheckShape( |
no test coverage detected