• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("@flatbuffers//:build_defs.bzl", "flatbuffer_android_library", "flatbuffer_cc_library", "flatbuffer_java_library", "flatbuffer_py_library")
2
3package(
4    default_visibility = [
5        "//visibility:public",
6    ],
7    licenses = ["notice"],  # Apache 2.0
8)
9
10exports_files(["metadata_schema.fbs"])
11
12flatbuffer_py_library(
13    name = "schema_py",
14    srcs = ["@org_tensorflow//tensorflow/lite/schema:schema.fbs"],
15)
16
17# Generic schema for inference on device.
18flatbuffer_android_library(
19    name = "schema_fbs_android",
20    srcs = ["@org_tensorflow//tensorflow/lite/schema:schema.fbs"],
21    custom_package = "org.tensorflow.lite.schema",
22)
23
24flatbuffer_java_library(
25    name = "schema_fbs_java",
26    srcs = ["@org_tensorflow//tensorflow/lite/schema:schema.fbs"],
27    custom_package = "org.tensorflow.lite.schema",
28)
29
30# Generic schema for model metadata.
31flatbuffer_cc_library(
32    name = "metadata_schema_cc",
33    srcs = ["metadata_schema.fbs"],
34)
35
36flatbuffer_py_library(
37    name = "metadata_schema_py",
38    srcs = ["metadata_schema.fbs"],
39)
40
41flatbuffer_java_library(
42    name = "metadata_schema_java",
43    srcs = ["metadata_schema.fbs"],
44    custom_package = "org.tensorflow.lite.support.metadata.schema",
45)
46
47flatbuffer_android_library(
48    name = "metadata_schema_fbs_android",
49    srcs = ["metadata_schema.fbs"],
50    custom_package = "org.tensorflow.lite.support.metadata.schema",
51)
52