examples/tflm/soundstream/soundstream_decoder.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/decoder.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
    decoder_tensor_arena[kelvin::soundstream::decoder::kTensorArenaSizeBytes]
    __attribute__((aligned(64)));
}  // namespace

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

  TfLiteTensor *decoder_input = decoder->interpreter->input(0);
  TfLiteTensor *decoder_output = decoder->interpreter->output(0);

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

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