xmos-ai-tools 1.3.2.dev80__py3-none-macosx_10_15_universal2.whl

xmos_ai_tools/runtime/include/signal/micro/kernels/filter_bank_spectral_subtraction_flexbuffers_generated_data.h

@@ -0,0 +1,26 @@
+/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_FILTER_BANK_SPECTRAL_SUBTRACTION_FLEXBUFFERS_DATA_H_
+#define SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_FILTER_BANK_SPECTRAL_SUBTRACTION_FLEXBUFFERS_DATA_H_
+
+extern const int g_gen_data_size_filter_bank_spectral_subtraction_32_channel;
+extern const unsigned char
+    g_gen_data_filter_bank_spectral_subtraction_32_channel[];
+extern const int g_gen_data_size_filter_bank_spectral_subtraction_16_channel;
+extern const unsigned char
+    g_gen_data_filter_bank_spectral_subtraction_16_channel[];
+
+#endif  // SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_FILTER_BANK_SPECTRAL_SUBTRACTION_FLEXBUFFERS_DATA_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/framer_flexbuffers_generated_data.h

@@ -0,0 +1,25 @@
+/* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_FRAMER_FLEXBUFFERS_DATA_H_
+#define SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_FRAMER_FLEXBUFFERS_DATA_H_
+
+extern const int g_gen_data_size_3_1_0_framer;
+extern const unsigned char g_gen_data_3_1_0_framer[];
+
+extern const int g_gen_data_size_5_2_1_framer;
+extern const unsigned char g_gen_data_5_2_1_framer[];
+
+#endif  // SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_FRAMER_FLEXBUFFERS_DATA_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/irfft.h

@@ -0,0 +1,31 @@
+/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+#ifndef SIGNAL_MICRO_KERNELS_IRFFT_H_
+#define SIGNAL_MICRO_KERNELS_IRFFT_H_
+
+#include "tensorflow/lite/micro/micro_common.h"
+
+namespace tflite {
+namespace tflm_signal {
+
+TFLMRegistration* Register_IRFFT();
+TFLMRegistration* Register_IRFFT_FLOAT();
+TFLMRegistration* Register_IRFFT_INT16();
+TFLMRegistration* Register_IRFFT_INT32();
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_MICRO_KERNELS_IRFFT_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/overlap_add_flexbuffers_generated_data.h

@@ -0,0 +1,25 @@
+/* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_OVERLAP_ADD_FLEXBUFFERS_DATA_H_
+#define SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_OVERLAP_ADD_FLEXBUFFERS_DATA_H_
+
+extern const int g_gen_data_size_overlap_add_float;
+extern const unsigned char g_gen_data_overlap_add_float[];
+
+extern const int g_gen_data_size_overlap_add_int16;
+extern const unsigned char g_gen_data_overlap_add_int16[];
+
+#endif  // SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_OVERLAP_ADD_FLEXBUFFERS_DATA_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/pcan_flexbuffers_generated_data.h

@@ -0,0 +1,7 @@
+#ifndef SIGNAL_MICRO_KERNELS_PCAN_FLEXBUFFERS_GENERATED_DATA_H_
+#define SIGNAL_MICRO_KERNELS_PCAN_FLEXBUFFERS_GENERATED_DATA_H_
+
+extern const int g_gen_data_size_snr_shift_6_test;
+extern const unsigned char g_gen_data_snr_shift_6_test[];
+
+#endif  // SIGNAL_MICRO_KERNELS_PCAN_FLEXBUFFERS_GENERATED_DATA_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/rfft.h

@@ -0,0 +1,31 @@
+/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+#ifndef SIGNAL_MICRO_KERNELS_RFFT_H_
+#define SIGNAL_MICRO_KERNELS_RFFT_H_
+
+#include "tensorflow/lite/micro/micro_common.h"
+
+namespace tflite {
+namespace tflm_signal {
+
+TFLMRegistration* Register_RFFT();
+TFLMRegistration* Register_RFFT_FLOAT();
+TFLMRegistration* Register_RFFT_INT16();
+TFLMRegistration* Register_RFFT_INT32();
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_MICRO_KERNELS_RFFT_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/stacker_flexbuffers_generated_data.h

@@ -0,0 +1,25 @@
+/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_STACKER_FLEXBUFFERS_DATA_H_
+#define SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_STACKER_FLEXBUFFERS_DATA_H_
+
+extern const int g_gen_data_size_stacker_3_channels_step_1;
+extern const unsigned char g_gen_data_stacker_3_channels_step_1[];
+
+extern const int g_gen_data_size_stacker_10_channels_step_2;
+extern const unsigned char g_gen_data_stacker_10_channels_step_2[];
+
+#endif  // SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_STACKER_FLEXBUFFERS_DATA_H_

xmos_ai_tools/runtime/include/signal/micro/kernels/window_flexbuffers_generated_data.h

@@ -0,0 +1,25 @@
+/* Copyright 2020 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_WINDOW_FLEXBUFFERS_DATA_H_
+#define SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_WINDOW_FLEXBUFFERS_DATA_H_
+
+extern const int g_gen_data_size_window_shift_12;
+extern const unsigned char g_gen_data_window_shift_12[];
+
+extern const int g_gen_data_size_window_shift_8;
+extern const unsigned char g_gen_data_window_shift_8[];
+
+#endif  // SIGNAL_MICRO_KERNELS_TEST_DATA_GENERATION_GENERATE_WINDOW_FLEXBUFFERS_DATA_H_

xmos_ai_tools/runtime/include/signal/src/circular_buffer.h

@@ -0,0 +1,118 @@
+/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_CIRCULAR_BUFFER_H_
+#define SIGNAL_SRC_CIRCULAR_BUFFER_H_
+
+#include <stddef.h>
+#include <stdint.h>
+
+namespace tflite {
+namespace tflm_signal {
+// TODO(b/286250473): remove namespace once de-duped libraries above
+struct CircularBuffer {
+  // Max number of elements, value passed-in to CircularBufferAlloc.
+  size_t capacity;
+  // Next position to read.
+  size_t read;
+  // Next position to write.
+  size_t write;
+  // Flag to indicate emptiness.
+  int32_t empty;
+  // Auto-generated size variable
+  int32_t buffer_size;
+  // Array of the circular buffer elements (integers).
+  int16_t* buffer;
+};
+
+// Returns the size of the memory that the circular buffer needs
+// in order to hold `capacity` items.
+size_t CircularBufferGetNeededMemory(size_t capacity);
+
+// Initialize an instance of the circular buffer that holds `capacity` items.
+// `state` points to a memory allocation of size `state_size`. The size
+//  should be greater or equal to the value returned by
+//  CircularBufferGetNeededMemory(capacity). Fails if it isn't.
+//  On success, returns a pointer to the circular buffer's object.
+CircularBuffer* CircularBufferInit(size_t capacity, void* state,
+                                   size_t state_size);
+
+// Reset a circular buffer to its initial empty state
+void CircularBufferReset(CircularBuffer* cb);
+
+size_t CircularBufferCapacity(const CircularBuffer* cb);
+
+bool CircularBufferFull(const CircularBuffer* cb);
+
+bool CircularBufferEmpty(const CircularBuffer* cb);
+
+// Returns the number of elements ready to read
+size_t CircularBufferAvailable(const CircularBuffer* cb);
+
+// Returns the number of elements available to write.
+size_t CircularBufferCanWrite(const CircularBuffer* cb);
+
+// Adds a single `value` to the buffer and advances the write pointer.
+void CircularBufferAdd(CircularBuffer* cb, int16_t value);
+
+// Writes `n` `values` into the buffer and advances the write pointer.
+void CircularBufferWrite(CircularBuffer* cb, const int16_t* values, size_t n);
+
+// Writes `n` zeros into the buffer and advances the write pointer.
+void CircularBufferWriteZeros(CircularBuffer* cb, size_t n);
+
+// Returns a pointer to a buffer where elements can be written, and
+// advances the write pointer as though they have already been written.
+// Fails if `n` elements are not available contiguously at the current
+// write position.
+int16_t* CircularBufferReserveForWrite(CircularBuffer* cb, size_t n);
+
+// Copies the final region (`count` elements) of the buffer `n` times, to
+// the end of the buffer.
+void CircularBufferExtend(CircularBuffer* cb, size_t count, int32_t n);
+
+// Reads a single value from the buffer and advances the read pointer
+int16_t CircularBufferRemove(CircularBuffer* cb);
+
+// Reads the value at the given `index`, does not modify the read pointer.
+int16_t CircularBufferPeek(const CircularBuffer* cb, size_t index);
+
+// Rewinds to restore the previous `n` values read
+void CircularBufferRewind(CircularBuffer* cb, size_t n);
+
+// Returns a pointer directly into the circular buffer at the given `index`.
+// Caller is responsible for not reading past the end.
+const int16_t* CircularBufferPeekDirect(const CircularBuffer* cb, size_t index);
+
+// Returns a pointer into the circular buffer at the current read pointer,
+// setting `n` to the number of values available to be read from here.
+const int16_t* CircularBufferPeekMax(const CircularBuffer* cb, size_t* n);
+
+// Copies `n` `values` from the buffer and does not advance the read
+// pointer and does not update the empty flag.
+void CircularBufferGet(CircularBuffer* cb, size_t n, int16_t* values);
+
+// Discards the next `n` values by advancing the read index.
+// Valid for n > 0.
+void CircularBufferDiscard(CircularBuffer* cb, size_t n);
+
+// Shifts the buffer with `n` values (`n` can be negative) by moving
+// the read index.
+void CircularBufferShift(CircularBuffer* cb, int n);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_SRC_CIRCULAR_BUFFER_H_

xmos_ai_tools/runtime/include/signal/src/complex.h

@@ -0,0 +1,29 @@
+/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_COMPLEX_H_
+#define SIGNAL_SRC_COMPLEX_H_
+
+#include <stdint.h>
+
+// We would use the standard complex type in complex.h, but there's
+// no guarantee that all architectures will support it.
+template <typename T>
+struct Complex {
+  T real;
+  T imag;
+};
+
+#endif  // SIGNAL_SRC_COMPLEX_H_

xmos_ai_tools/runtime/include/signal/src/energy.h

@@ -0,0 +1,38 @@
+/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_ENERGY_H_
+#define SIGNAL_ENERGY_H_
+
+#include <stdint.h>
+
+#include "signal/src/complex.h"
+
+namespace tflite {
+namespace tflm_signal {
+// TODO(b/286250473): remove namespace once de-duped libraries above
+
+// Calculates the power spectrum from a DFT output between start and end indices
+//
+// * `start_index` and `end_index` must valid indices into `input`
+// * `output` must be the same size as `input`. Only the values at indices
+//   `start_index` and `end_index` inclusive should be considered valid.
+void SpectrumToEnergy(const Complex<int16_t>* input, int start_index,
+                      int end_index, uint32_t* output);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_ENERGY_H_

xmos_ai_tools/runtime/include/signal/src/fft_auto_scale.h

@@ -0,0 +1,35 @@
+/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_FFT_AUTO_SCALE_H_
+#define SIGNAL_SRC_FFT_AUTO_SCALE_H_
+
+#include <stddef.h>
+#include <stdint.h>
+
+// TODO(b/286250473): remove namespace once de-duped libraries
+namespace tflite {
+namespace tflm_signal {
+
+// Auto scales `input` and write the result to `output`
+// Elements in `input` are left shifted to maximize the amplitude without
+// clipping,
+// * both `input` and `output` must be of size `size`
+int FftAutoScale(const int16_t* input, int size, int16_t* output);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_SRC_FFT_AUTO_SCALE_H_

xmos_ai_tools/runtime/include/signal/src/filter_bank.h

@@ -0,0 +1,69 @@
+/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_FILTER_BANK_H_
+#define SIGNAL_SRC_FILTER_BANK_H_
+
+#include <stdint.h>
+
+namespace tflite {
+namespace tflm_signal {
+// TODO(b/286250473): remove namespace once de-duped libraries above
+
+struct FilterbankConfig {
+  // Number of filterbank channels
+  int32_t num_channels;
+
+  // Each of the following three arrays is of size num_channels + 1
+  // An extra channel is needed for scratch. See implementation of
+  // FilterbankAccumulateChannels() for more details
+
+  // For each channel, the index in the input (spectrum) where its band starts
+  const int16_t* channel_frequency_starts;
+  // For each channel, the index in the weights/unweights arrays where
+  // it filter weights start
+  const int16_t* channel_weight_starts;
+  // For each channel, the number of bins in the input (spectrum) that span
+  // its band
+  const int16_t* channel_widths;
+
+  // The weights array holds the triangular filter weights of all the filters
+  // in the bank. The output of each filter in the bank is caluclated by
+  // multiplying the elements in the input spectrum that are in its band
+  // (see above: channel_frequency_starts, channel_widths) by the filter weights
+  // then accumulating. Each element in the unweights array holds the 1 minus
+  // corresponding elements in the weights array and is used to make this
+  // operation more efficient. For more details, see documnetation in
+  // FilterbankAccumulateChannels()
+  const int16_t* weights;
+  const int16_t* unweights;
+  int32_t output_scale;
+
+  int32_t input_correction_bits;
+};
+
+// Accumulate the energy spectrum bins in `input` into filter bank channels
+// contained in `output`.
+// * `input` - Spectral energy array
+// * `output` - of size `config.num_channels` + 1.
+//              Elements [1:num_channels] contain the filter bank channels.
+//              Element 0 is used as scratch and should be ignored
+void FilterbankAccumulateChannels(const FilterbankConfig* config,
+                                  const uint32_t* input, uint64_t* output);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_SRC_FILTER_BANK_H_

xmos_ai_tools/runtime/include/signal/src/filter_bank_log.h

@@ -0,0 +1,38 @@
+/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_FILTER_BANK_LOG_H_
+#define SIGNAL_SRC_FILTER_BANK_LOG_H_
+
+#include <stdint.h>
+
+namespace tflite {
+namespace tflm_signal {
+// TODO(b/286250473): remove namespace once de-duped libraries above
+
+// Apply natural log to each element in array `input` of size `num_channels`
+// with pre-shift and post scaling.
+// The operation is roughly equivalent to:
+// `output` = min(Log(`input` << `correction_bits`) * `output_scale`, INT16_MAX)
+//  Where:
+//    If (input << `correction_bits`) is 1 or 0, the function returns 0
+void FilterbankLog(const uint32_t* input, int num_channels,
+                   int32_t output_scale, uint32_t correction_bits,
+                   int16_t* output);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_SRC_FILTER_BANK_LOG_H_

xmos_ai_tools/runtime/include/signal/src/filter_bank_spectral_subtraction.h

@@ -0,0 +1,73 @@
+/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_FILTER_BANK_SPECTRAL_SUBTRACTION_H_
+#define SIGNAL_SRC_FILTER_BANK_SPECTRAL_SUBTRACTION_H_
+
+#include <stdint.h>
+
+namespace tflite {
+namespace tflm_signal {
+// TODO(b/286250473): remove namespace once de-duped libraries above
+
+struct SpectralSubtractionConfig {
+  // Number of filterbank channels in input and output
+  int32_t num_channels;
+  // The constant used for the lowpass filter for finding the noise.
+  // Higher values correspond to more aggressively adapting estimates
+  // of the noise.
+  // Scale is 1 << spectral_subtraction_bits
+  uint32_t smoothing;
+  // One minus smoothing constant for low pass filter.
+  // Scale is 1 << spectral_subtraction_bits
+  uint32_t one_minus_smoothing;
+  // The maximum cap to subtract away from the signal (ie, if this is
+  // 0.2, then the result of spectral subtraction will not go below
+  // 0.2 * signal).
+  //  Scale is 1 << spectral_subtraction_bits
+  uint32_t min_signal_remaining;
+  // If positive, specifies the filter coefficient for odd-index
+  // channels, while 'smoothing' is used as the coefficient for even-
+  // index channels. Otherwise, the same filter coefficient is
+  // used on all channels.
+  //  Scale is 1 << spectral_subtraction_bits
+  uint32_t alternate_smoothing;
+  // Alternate One minus smoothing constant for low pass filter.
+  // Scale is 1 << spectral_subtraction_bits
+  uint32_t alternate_one_minus_smoothing;
+  // Extra fractional bits for the noise_estimate smoothing filter.
+  uint32_t smoothing_bits;
+  // Scaling bits for some members of this struct
+  uint32_t spectral_subtraction_bits;
+  // If true, when the filterbank level drops below the output,
+  // the noise estimate will be forced down to the new noise level.
+  // If false, the noise estimate will remain above the current
+  // filterbank output (but the subtraction will still keep the
+  // output non negative).
+  bool clamping;
+};
+
+// Apply spectral subtraction to each element in `input`, then write the result
+// to `output` and `noise_estimate`. `input`, `output` and `noise estimate`
+// must all be of size `config.num_channels`. `config` holds the
+//  parameters of the spectral subtraction algorithm.
+void FilterbankSpectralSubtraction(const SpectralSubtractionConfig* config,
+                                   const uint32_t* input, uint32_t* output,
+                                   uint32_t* noise_estimate);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_SRC_FILTER_BANK_SPECTRAL_SUBTRACTION_H_

xmos_ai_tools/runtime/include/signal/src/filter_bank_square_root.h

@@ -0,0 +1,34 @@
+/* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
+
+#ifndef SIGNAL_SRC_FILTER_BANK_SQUARE_ROOT_H_
+#define SIGNAL_SRC_FILTER_BANK_SQUARE_ROOT_H_
+
+#include <stdint.h>
+
+namespace tflite {
+namespace tflm_signal {
+// TODO(b/286250473): remove namespace once de-duped libraries above
+
+// Apply square root to each element in `input`, then shift right by
+// `scale_down_bits` before writing the result to `output`,
+// `input` and `output` must both be of size `num_channels`
+void FilterbankSqrt(const uint64_t* input, int num_channels,
+                    int scale_down_bits, uint32_t* output);
+
+}  // namespace tflm_signal
+}  // namespace tflite
+
+#endif  // SIGNAL_SRC_FILTER_BANK_SQUARE_ROOT_H_