mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-01 17:57:27 +00:00
ba753cef3e
- Upgraded the Bazel BuildTool to the latest version (8.2.1). - Updated the `tflm-ci` Docker image and pushed the new tag (0.6.1) to the `ghcr.io/tflm-bot/tflm-ci registry`. - Updated the WORKSPACE to load `rules_cc` and `rules_shell` explicitly - Ran `buildifier` to ensure all BUILD files to have all the fixes. BUG=Clean-up
54 lines
1.3 KiB
Text
54 lines
1.3 KiB
Text
load("@rules_cc//cc:cc_binary.bzl", "cc_binary")
|
|
load(
|
|
"//tensorflow/lite/micro:build_def.bzl",
|
|
"generate_cc_arrays",
|
|
"tflm_cc_library",
|
|
)
|
|
|
|
package(
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
generate_cc_arrays(
|
|
name = "generated_simple_add_model_cc",
|
|
src = "models/simple_add_model.tflite",
|
|
out = "models/simple_add_model_model_data.cc",
|
|
)
|
|
|
|
generate_cc_arrays(
|
|
name = "generated_simple_add_model_hdr",
|
|
src = "models/simple_add_model.tflite",
|
|
out = "models/simple_add_model_model_data.h",
|
|
)
|
|
|
|
tflm_cc_library(
|
|
name = "simple_add_model_data",
|
|
srcs = [
|
|
":generated_simple_add_model_cc",
|
|
],
|
|
hdrs = [
|
|
":generated_simple_add_model_hdr",
|
|
],
|
|
)
|
|
|
|
cc_binary(
|
|
name = "baseline_memory_footprint",
|
|
srcs = ["baseline_memory_footprint.cc"],
|
|
deps = [
|
|
"//tensorflow/lite/c:common",
|
|
"//tensorflow/lite/micro:system_setup",
|
|
],
|
|
)
|
|
|
|
cc_binary(
|
|
name = "interpreter_memory_footprint",
|
|
srcs = ["interpreter_memory_footprint.cc"],
|
|
deps = [
|
|
":simple_add_model_data",
|
|
"//tensorflow/lite/c:common",
|
|
"//tensorflow/lite/micro:micro_profiler",
|
|
"//tensorflow/lite/micro:op_resolvers",
|
|
"//tensorflow/lite/micro:system_setup",
|
|
"//tensorflow/lite/micro/benchmarks:micro_benchmark",
|
|
],
|
|
)
|