examples/tflm/soundstream/soundstream_encoder.cc

// Copyright 2023 Google LLC
// Licensed under the Apache License, Version 2.0, see LICENSE for details.
// SPDX-License-Identifier: Apache-2.0

#include "examples/tflm/soundstream/encoder.h"

typedef struct {
  uint32_t return_code;  // Populated in kelvin_start
  uint32_t output_ptr;
  uint32_t length;
} OutputHeader;

__attribute__((section(".model_output_header"))) OutputHeader output_header = {
    .output_ptr = 0,
    .length = 0,
};

namespace {
uint8_t
    encoder_tensor_arena[kelvin::soundstream::encoder::kTensorArenaSizeBytes]
    __attribute__((aligned(64)));
}  // namespace

int main(int argc, char **argv) {
  auto encoder = kelvin::soundstream::encoder::Setup(encoder_tensor_arena);
  if (!encoder) {
    MicroPrintf("Unable to construct encoder");
    return -1;
  }

  TfLiteTensor *encoder_input = encoder->interpreter->input(0);
  TfLiteTensor *encoder_output = encoder->interpreter->output(0);

  memset(encoder_input->data.uint8, 0, encoder_input->bytes);
  TfLiteStatus invoke_status = encoder->interpreter->Invoke();
  if (invoke_status != kTfLiteOk) {
    MicroPrintf("Failed to invoke encoder");
    return -1;
  }

  output_header.length = encoder_output->bytes;
  output_header.output_ptr =
      reinterpret_cast<uint32_t>(encoder_output->data.uint8);
  return 0;
}