Skip to content

Commit

Permalink
style fix
Browse files Browse the repository at this point in the history
  • Loading branch information
turbotoribio committed Jul 12, 2023
1 parent aa9df4c commit 22a36e3
Showing 1 changed file with 13 additions and 17 deletions.
30 changes: 13 additions & 17 deletions signal/micro/kernels/delay.cc
Original file line number Diff line number Diff line change
Expand Up @@ -53,8 +53,7 @@ void* Init(TfLiteContext* context, const char* buffer, size_t length) {
return nullptr;
}

FlexbufferWrapper fbw(reinterpret_cast<const uint8_t*>(buffer),
length);
FlexbufferWrapper fbw(reinterpret_cast<const uint8_t*>(buffer), length);
params->delay_length = fbw.ElementAsInt32(kDelayLengthIndex);
return params;
}
Expand Down Expand Up @@ -89,21 +88,19 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {
context, params->outer_dims * sizeof(int8_t*)));
params->circular_buffers = static_cast<tflm_signal::CircularBuffer**>(
context->AllocatePersistentBuffer(
context,
params->outer_dims * sizeof(tflm_signal::CircularBuffer*)));
context, params->outer_dims * sizeof(tflm_signal::CircularBuffer*)));

for (int i = 0; i < params->outer_dims; i++) {
size_t capacity = params->frame_size + params->delay_length;

size_t state_size =
tflm_signal::CircularBufferGetNeededMemory(capacity);
size_t state_size = tflm_signal::CircularBufferGetNeededMemory(capacity);
params->state_buffers[i] =
static_cast<int8_t*>(context->AllocatePersistentBuffer(
context, state_size * sizeof(int8_t)));
params->circular_buffers[i] = tflm_signal::CircularBufferInit(
capacity, params->state_buffers[i], state_size);
tflm_signal::CircularBufferWriteZeros(params->circular_buffers[i],
params->delay_length);
params->delay_length);
}

micro_context->DeallocateTempTfLiteTensor(input);
Expand All @@ -116,22 +113,21 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
reinterpret_cast<TFLMSignalFrontendDelayParams*>(node->user_data);
const TfLiteEvalTensor* input =
micro::GetEvalInput(context, node, kInputTensor);
TfLiteEvalTensor* output =
micro::GetEvalOutput(context, node, kOutputTensor);
TfLiteEvalTensor* output = micro::GetEvalOutput(context, node, kOutputTensor);

const int16_t* input_data = micro::GetTensorData<int16_t>(input);
int16_t* output_data = micro::GetTensorData<int16_t>(output);

for (int dim_index = 0, sample_index = 0; dim_index < params->outer_dims;
dim_index++, sample_index += params->frame_size) {
tflm_signal::CircularBufferWrite(
params->circular_buffers[dim_index], &input_data[sample_index],
params->frame_size);
tflm_signal::CircularBufferWrite(params->circular_buffers[dim_index],
&input_data[sample_index],
params->frame_size);
tflm_signal::CircularBufferGet(params->circular_buffers[dim_index],
params->frame_size,
&output_data[sample_index]);
tflm_signal::CircularBufferDiscard(
params->circular_buffers[dim_index], params->frame_size);
params->frame_size,
&output_data[sample_index]);
tflm_signal::CircularBufferDiscard(params->circular_buffers[dim_index],
params->frame_size);
}
return kTfLiteOk;
}
Expand All @@ -141,7 +137,7 @@ void Reset(TfLiteContext* context, void* buffer) {
for (int i = 0; i < params->outer_dims; ++i) {
tflm_signal::CircularBufferReset(params->circular_buffers[i]);
tflm_signal::CircularBufferWriteZeros(params->circular_buffers[i],
params->delay_length);
params->delay_length);
}
}

Expand Down

0 comments on commit 22a36e3

Please sign in to comment.