blob: 1805e488ac751d9bbb08d11dc57e2f9b47d6ffa4 [file] [log] [blame]
// Copyright 2020 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <cstdint>
#include <brillo/brillo_export.h>
// The following two functions are exported as test driver for running
// arbitrary tflite models within mlservice.
// Both function names and signatures should not be changed.
// Parses config_bytes as CrOSBenchmarkConfig and runs mlservice inference
// accordingly.
extern "C" BRILLO_EXPORT int32_t benchmark_start(const void* config_bytes,
int32_t config_bytes_size,
void** results_bytes,
int32_t* results_bytes_size);
// Deletes results_bytes.
extern "C" BRILLO_EXPORT void free_benchmark_results(void* results_bytes);
#endif // ML_BENCHMARK_H_