diff options
Diffstat (limited to 'tensorflow_lite_support/metadata/BUILD')
-rw-r--r-- | tensorflow_lite_support/metadata/BUILD | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/tensorflow_lite_support/metadata/BUILD b/tensorflow_lite_support/metadata/BUILD new file mode 100644 index 00000000..db69bd25 --- /dev/null +++ b/tensorflow_lite_support/metadata/BUILD @@ -0,0 +1,51 @@ +load("@flatbuffers//:build_defs.bzl", "flatbuffer_android_library", "flatbuffer_cc_library", "flatbuffer_java_library", "flatbuffer_py_library") + +package( + default_visibility = [ + "//visibility:public", + ], + licenses = ["notice"], # Apache 2.0 +) + +exports_files(["metadata_schema.fbs"]) + +flatbuffer_py_library( + name = "schema_py", + srcs = ["@org_tensorflow//tensorflow/lite/schema:schema.fbs"], +) + +# Generic schema for inference on device. +flatbuffer_android_library( + name = "schema_fbs_android", + srcs = ["@org_tensorflow//tensorflow/lite/schema:schema.fbs"], + custom_package = "org.tensorflow.lite.schema", +) + +flatbuffer_java_library( + name = "schema_fbs_java", + srcs = ["@org_tensorflow//tensorflow/lite/schema:schema.fbs"], + custom_package = "org.tensorflow.lite.schema", +) + +# Generic schema for model metadata. +flatbuffer_cc_library( + name = "metadata_schema_cc", + srcs = ["metadata_schema.fbs"], +) + +flatbuffer_py_library( + name = "metadata_schema_py", + srcs = ["metadata_schema.fbs"], +) + +flatbuffer_java_library( + name = "metadata_schema_java", + srcs = ["metadata_schema.fbs"], + custom_package = "org.tensorflow.lite.support.metadata.schema", +) + +flatbuffer_android_library( + name = "metadata_schema_fbs_android", + srcs = ["metadata_schema.fbs"], + custom_package = "org.tensorflow.lite.support.metadata.schema", +) |