janitor

janitor
git clone https://git.prime8.dev/janitor.git
Log | Files | Refs | README | LICENSE

rfft.cc (1164B)

      1 /**
      2  * Copyright (c)  2022-2023  Xiaomi Corporation (authors: Fangjun Kuang)
      3  *
      4  * See LICENSE for clarification regarding multiple authors
      5  *
      6  * Licensed under the Apache License, Version 2.0 (the "License");
      7  * you may not use this file except in compliance with the License.
      8  * You may obtain a copy of the License at
      9  *
     10  *     http://www.apache.org/licenses/LICENSE-2.0
     11  *
     12  * Unless required by applicable law or agreed to in writing, software
     13  * distributed under the License is distributed on an "AS IS" BASIS,
     14  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     15  * See the License for the specific language governing permissions and
     16  * limitations under the License.
     17  */
     18 
     19 #include "kaldi-native-fbank/python/csrc/rfft.h"
     20 
     21 #include <vector>
     22 #include <cstdint>
     23 
     24 #include "kaldi-native-fbank/csrc/rfft.h"
     25 
     26 namespace knf {
     27 
     28 void PybindRfft(py::module &m) {  // NOLINT
     29   py::class_<Rfft>(m, "Rfft")
     30       .def(py::init<int32_t>(), py::arg("n"))
     31       .def("compute",
     32            [](Rfft &self, std::vector<float> &d) -> std::vector<float> {
     33              self.Compute(d.data());
     34              return d;
     35            });
     36 }
     37 
     38 }  // namespace knf