wehub-resource-sync
163 lines
4.1 KiB
Python
163 lines
4.1 KiB
Python
load("@flatbuffers//:build_defs.bzl", "flatbuffer_cc_library")
|
|
load("@rules_cc//cc:cc_library.bzl", "cc_library")
|
|
|
|
# copybara:uncomment load("@flatbuffers//:flatbuffers.bzl", "flatbuffers_library", "ts_flatbuffers_library")
|
|
load("//tensorflow:tensorflow.bzl", "tf_cc_test")
|
|
load("//tensorflow:tensorflow.default.bzl", "get_compatible_with_portable")
|
|
# copybara:uncomment load("//tools/build_defs/cc:cc_embed_data.bzl", "cc_embed_data")
|
|
|
|
package(
|
|
# copybara:uncomment default_applicable_licenses = ["//tensorflow:LICENSE"],
|
|
default_visibility = [
|
|
"//visibility:public",
|
|
],
|
|
licenses = ["notice"],
|
|
)
|
|
|
|
exports_files(
|
|
srcs = [
|
|
"conversion_metadata.fbs",
|
|
"debug_metadata.fbs",
|
|
"schema.fbs",
|
|
"schema_v0.fbs",
|
|
"schema_v1.fbs",
|
|
"schema_v2.fbs",
|
|
"schema_v3.fbs",
|
|
"schema_v3a.fbs",
|
|
"schema_v3b.fbs",
|
|
"schema_v3c.fbs",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "tflite_internal_cc_3p_api_deps_src",
|
|
srcs = [
|
|
":debug_metadata_fbs_srcs",
|
|
":schema_fbs_srcs",
|
|
":schema_utils.h",
|
|
],
|
|
visibility = ["//tensorflow/lite:__pkg__"],
|
|
)
|
|
|
|
flatbuffer_cc_library(
|
|
name = "schema_fbs",
|
|
srcs = ["schema.fbs"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
gen_reflections = True,
|
|
)
|
|
|
|
# copybara:uncomment_begin(google-only)
|
|
# cc_embed_data(
|
|
# name = "schema_reflection_data",
|
|
# srcs = ["schema.bfbs"],
|
|
# outs = [
|
|
# "schema_reflection_data.cc",
|
|
# "schema_reflection_data.h",
|
|
# "schema_reflection_data.o",
|
|
# ],
|
|
# embedopts = ["--namespace=tflite"],
|
|
# )
|
|
# copybara:uncomment_end
|
|
|
|
# Generic schema for flatbuffer converter (but with mutable makes bigger).
|
|
flatbuffer_cc_library(
|
|
name = "schema_fbs_with_mutable",
|
|
srcs = ["schema.fbs"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
flatc_args = [
|
|
"--gen-mutable",
|
|
"--gen-object-api",
|
|
],
|
|
out_prefix = "mutable/",
|
|
)
|
|
|
|
flatbuffer_cc_library(
|
|
name = "debug_metadata_fbs",
|
|
srcs = ["debug_metadata.fbs"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
)
|
|
|
|
flatbuffer_cc_library(
|
|
name = "debug_metadata_fbs_with_mutable",
|
|
srcs = ["debug_metadata.fbs"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
flatc_args = [
|
|
"--gen-mutable",
|
|
"--gen-object-api",
|
|
],
|
|
out_prefix = "mutable/",
|
|
)
|
|
|
|
# Generic schema for inference on device (but with reflections makes bigger).
|
|
flatbuffer_cc_library(
|
|
name = "schema_fbs_with_reflection",
|
|
srcs = ["schema.fbs"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
flatc_args = [
|
|
"--reflect-types",
|
|
"--reflect-names",
|
|
"--no-union-value-namespacing",
|
|
"--gen-object-api",
|
|
],
|
|
out_prefix = "reflection/",
|
|
)
|
|
|
|
cc_library(
|
|
name = "schema_utils",
|
|
srcs = ["schema_utils.cc"],
|
|
hdrs = ["schema_utils.h"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
# visibility = [":utils_friends"],
|
|
deps = [
|
|
":schema_fbs",
|
|
"//tensorflow/compiler/mlir/lite/kernels/internal:compatibility_macros",
|
|
"@flatbuffers//:runtime_cc",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "schema_conversion_utils",
|
|
srcs = ["schema_conversion_utils.cc"],
|
|
hdrs = ["schema_conversion_utils.h"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
# visibility = [":utils_friends"],
|
|
deps = [
|
|
":schema_fbs",
|
|
"@flatbuffers",
|
|
],
|
|
)
|
|
|
|
# Schema test to make sure we don't introduce backward incompatible changes
|
|
# to schemas.
|
|
tf_cc_test(
|
|
name = "flatbuffer_compatibility_test",
|
|
size = "small",
|
|
srcs = ["flatbuffer_compatibility_test.cc"],
|
|
data = [
|
|
"schema.fbs",
|
|
"schema_v3c.fbs",
|
|
],
|
|
tags = [
|
|
"no_oss",
|
|
"tflite_not_portable_android",
|
|
"tflite_not_portable_ios",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core/platform",
|
|
"@com_google_googletest//:gtest_main",
|
|
"@flatbuffers//:flatc_library",
|
|
],
|
|
)
|
|
|
|
# copybara:uncomment_begin(google-only)
|
|
# flatbuffers_library(
|
|
# name = "schema_fbslib",
|
|
# srcs = ["schema.fbs"],
|
|
# )
|
|
#
|
|
# ts_flatbuffers_library(
|
|
# name = "schema_ts_fbs",
|
|
# deps = [":schema_fbslib"],
|
|
# )
|
|
# copybara:uncomment_end
|