| 630 | } |
| 631 | |
| 632 | static void Fail(const Shape& shape, FftType type, |
| 633 | absl::Span<const int64> length, absl::string_view message) { |
| 634 | auto inferred_status = ShapeInference::InferFftShape(shape, type, length); |
| 635 | ASSERT_FALSE(inferred_status.ok()); |
| 636 | ASSERT_THAT(inferred_status.status().error_message(), |
| 637 | HasSubstr(std::string(message))); |
| 638 | } |
| 639 | |
| 640 | } // namespace fft |
| 641 |