| 622 | "innermost dimension matches fft_length/2+1"; |
| 623 | |
| 624 | static void Pass(const Shape& shape, FftType type, |
| 625 | absl::Span<const int64> length, const Shape& expected_shape) { |
| 626 | auto inferred_status = ShapeInference::InferFftShape(shape, type, length); |
| 627 | ASSERT_IS_OK(inferred_status.status()); |
| 628 | Shape inferred_shape = inferred_status.ValueOrDie(); |
| 629 | ASSERT_TRUE(ShapeUtil::Equal(inferred_shape, expected_shape)); |
| 630 | } |
| 631 | |
| 632 | static void Fail(const Shape& shape, FftType type, |
| 633 | absl::Span<const int64> length, absl::string_view message) { |