mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-03 10:47:33 +00:00
b766947f5a
The TFLM code_generator takes a TFLite model and generates C/C++ source code that can then be compiled into a binary. This change adds a Bazel macro for invoking the code generator and creating a cc_library with the resulting sources. It also updates the checked-in hello_world example with an appropriate BUILD file and updates the script to use Bazel instead of make. BUG=b/294230402
18 lines
581 B
Text
18 lines
581 B
Text
load("//tensorflow/lite/micro:build_def.bzl", "micro_copts")
|
|
|
|
package(default_visibility = ["//visibility:public"])
|
|
|
|
cc_library(
|
|
name = "micro_codegen_context",
|
|
srcs = ["micro_codegen_context.cc"],
|
|
hdrs = ["micro_codegen_context.h"],
|
|
copts = micro_copts(),
|
|
deps = [
|
|
"//tensorflow/lite/c:common",
|
|
"//tensorflow/lite/kernels:op_macros",
|
|
"//tensorflow/lite/kernels/internal:compatibility",
|
|
"//tensorflow/lite/micro:micro_context",
|
|
"//tensorflow/lite/micro:micro_graph",
|
|
"//tensorflow/lite/micro:micro_log",
|
|
],
|
|
)
|