| 4657 | class IRFftFunctor { |
| 4658 | public: |
| 4659 | Maybe<Tensor> operator()(const std::shared_ptr<one::Tensor>& input, int64_t n, int64_t dim, |
| 4660 | const Optional<std::string>& norm) const { |
| 4661 | std::string norm_str = norm.value_or("backward"); |
| 4662 | std::vector<int64_t> fft_dim{dim}; |
| 4663 | |
| 4664 | bool forward = false; |
| 4665 | fft_norm_mode norm_mode = fft_norm_mode::none; |
| 4666 | norm_mode = fft_norm_from_string(norm_str, forward); |
| 4667 | |
| 4668 | std::vector<int64_t> len{n}; |
| 4669 | return functional::FftC2R(input, len, fft_dim, static_cast<int32_t>(norm_mode), |
| 4670 | /*forward=*/forward, /*normalized=*/true); |
| 4671 | } |
| 4672 | }; |
| 4673 | |
| 4674 | class RFft2Functor { |
nothing calls this directly
no test coverage detected