mirror of
https://github.com/vee1e/tflite-micro.git
synced 2026-09-03 10:47:33 +00:00
807c35d6a9
* Work * Review update * Enabled tests * Bazel: Suppress warnings from external repositories * Fixed whl_test * Formatted * More format
40 lines
1 KiB
Text
40 lines
1 KiB
Text
load("@rules_cc//cc:cc_library.bzl", "cc_library")
|
|
load("//third_party/flatbuffers:build_defs.bzl", "flatbuffer_cc_library")
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//visibility:public",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
# Note: when wanting to generate the schema_generated.h, you must build as:
|
|
# bazel build schema_fbs_srcs.
|
|
flatbuffer_cc_library(
|
|
name = "schema_fbs",
|
|
srcs = ["//tensorflow/compiler/mlir/lite/schema:schema.fbs"],
|
|
)
|
|
|
|
# Generic schema for inference on device (but with reflections makes bigger).
|
|
flatbuffer_cc_library(
|
|
name = "schema_fbs_with_reflection",
|
|
srcs = ["//tensorflow/compiler/mlir/lite/schema:schema.fbs"],
|
|
flatc_args = [
|
|
"--reflect-types",
|
|
"--reflect-names",
|
|
"--no-union-value-namespacing",
|
|
"--gen-object-api",
|
|
],
|
|
out_prefix = "reflection/",
|
|
)
|
|
|
|
cc_library(
|
|
name = "schema_utils",
|
|
hdrs = [
|
|
"schema_generated.h",
|
|
"schema_utils.h",
|
|
],
|
|
deps = [
|
|
"//tensorflow/compiler/mlir/lite/schema:schema_utils",
|
|
],
|
|
)
|