Skip to content

module: tflm: rework module to use sink/source api #10804

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
softwarecki wants to merge 1 commit into
base: main
Choose a base branch
from
+19 −14
Open

module: tflm: rework module to use sink/source api #10804

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 19 additions & 14 deletions src/audio/tensorflow/tflm-classify.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,8 +3,9 @@
// Copyright(c) 2025 Intel Corporation. All rights reserved.

#include <sof/audio/module_adapter/module/generic.h>
#include <sof/audio/buffer.h>
#include <sof/audio/component.h>
#include <sof/audio/sink_api.h>
#include <sof/audio/source_api.h>
#include <sof/audio/data_blob.h>
#include <sof/audio/format.h>
#include <sof/audio/ipc-config.h>
Expand Down Expand Up @@ -174,16 +175,15 @@ int8_t expected_feature_yes[TFLM_FEATURE_SIZE] = {
*/

static int tflm_process(struct processing_module *mod,
struct input_stream_buffer *input_buffers,
int num_input_buffers,
struct output_stream_buffer *output_buffers,
int num_output_buffers)
struct sof_source **sources, int num_of_sources,
struct sof_sink **sinks, int num_of_sinks)
{
Comment on lines 177 to 180
struct tflm_comp_data *cd = module_get_private_data(mod);
struct comp_dev *dev = mod->dev;
struct audio_stream *source = input_buffers[0].data;
struct audio_stream *sink = output_buffers[0].data;
int features = input_buffers[0].size;
size_t frame_bytes = source_get_frame_bytes(sources[0]);
int features = source_get_data_frames_available(sources[0]);
const void *data_ptr, *buf_start;
size_t buf_size;
Comment on lines +185 to +186
int ret;

comp_dbg(dev, "entry");
Expand All @@ -192,12 +192,18 @@ static int tflm_process(struct processing_module *mod,
* by TFLM_FEATURE_SIZE until buffer empty.
*/
while (features >= TFLM_FEATURE_ELEM_COUNT) {
cd->tfc.audio_features = source->r_ptr;
ret = source_get_data(sources[0], TFLM_FEATURE_ELEM_COUNT * frame_bytes,
&data_ptr, &buf_start, &buf_size);
if (ret)
return ret;

cd->tfc.audio_features = data_ptr;
cd->tfc.audio_data_size = TFLM_FEATURE_ELEM_COUNT;
Comment on lines +195 to 201
ret = TF_ProcessClassify(&cd->tfc);
if (!ret) {
comp_err(dev, "classify failed %s.",
cd->tfc.error);
source_release_data(sources[0], 0);
return ret;
}

Expand All @@ -207,10 +213,9 @@ static int tflm_process(struct processing_module *mod,
cd->tfc.predictions[i], prediction[i]);
}

/* calc new free and available after moving onto next feature */
module_update_buffer_position(&input_buffers[0],
&output_buffers[0], TFLM_FEATURE_SIZE);
features = input_buffers[0].size;
/* advance by one stride */
source_release_data(sources[0], TFLM_FEATURE_SIZE * frame_bytes);
features = source_get_data_frames_available(sources[0]);
Comment on lines +216 to +218
}

return ret;
Expand All @@ -226,7 +231,7 @@ static int tflm_reset(struct processing_module *mod)
static const struct module_interface tflmcly_interface = {
.init = tflm_init,
// .prepare = tflm_prepare,
.process_audio_stream = tflm_process,
.process = tflm_process,
.set_configuration = tflm_set_config,
// .get_configuration = tflm_get_config,
.reset = tflm_reset,
Expand Down
Loading