aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorandroid-build-team Robot <android-build-team-robot@google.com>2021-06-21 14:30:52 +0000
committerandroid-build-team Robot <android-build-team-robot@google.com>2021-06-21 14:30:52 +0000
commit8969a758b21bd60e59811760e8166de65a058997 (patch)
tree63ba5a62524b5a035cee917ea4e3b7a06c6daf56
parentc2010feb97a01a856e2deaf96448660f813bde3f (diff)
parente3d3d057417bb8ad2baeadb9a3ffb77a2d5c7629 (diff)
downloadtflite-support-8969a758b21bd60e59811760e8166de65a058997.tar.gz
Change-Id: I6f55e470dcf0141fdf85b2cb5cdee81e0d415615
-rw-r--r--Android.bp98
-rw-r--r--OWNERS3
-rw-r--r--tensorflow_lite_support/custom_ops/kernel/sentencepiece/decoder_config.fbs2
-rw-r--r--tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs2
4 files changed, 103 insertions, 2 deletions
diff --git a/Android.bp b/Android.bp
new file mode 100644
index 00000000..4b012533
--- /dev/null
+++ b/Android.bp
@@ -0,0 +1,98 @@
+package {
+ default_applicable_licenses: ["external_tflite-support_license"],
+}
+
+// Added automatically by a large-scale-change that took the approach of
+// 'apply every license found to every target'. While this makes sure we respect
+// every license restriction, it may not be entirely correct.
+//
+// e.g. GPL in an MIT project might only apply to the contrib/ directory.
+//
+// Please consider splitting the single license below into multiple licenses,
+// taking care not to lose any license_kind information, and overriding the
+// default license using the 'licenses: [...]' property on targets as needed.
+//
+// For unused files, consider creating a 'fileGroup' with "//visibility:private"
+// to attach the license to, and including a comment whether the files may be
+// used in the current project.
+// See: http://go/android-license-faq
+license {
+ name: "external_tflite-support_license",
+ visibility: [":__subpackages__"],
+ license_kinds: [
+ "SPDX-license-identifier-Apache-2.0",
+ "SPDX-license-identifier-BSD",
+ "SPDX-license-identifier-MIT",
+ ],
+ license_text: [
+ "LICENSE",
+ ],
+}
+
+cc_library_static {
+ name: "tflite_support",
+ sdk_version: "current",
+ min_sdk_version: "30",
+ stl: "libc++_static",
+ export_include_dirs: ["."],
+ srcs: [
+ "tensorflow_lite_support/cc/utils/common_utils.cc",
+ "tensorflow_lite_support/custom_ops/kernel/sentencepiece/sentencepiece_tokenizer_tflite.cc",
+ "tensorflow_lite_support/custom_ops/kernel/sentencepiece/optimized_encoder.cc",
+ "tensorflow_lite_support/custom_ops/kernel/ragged/ragged_range_tflite.cc",
+ "tensorflow_lite_support/custom_ops/kernel/ragged/ragged_tensor_to_tensor_tflite.cc",
+ ],
+ cflags: [
+ "-Werror",
+ "-Wall",
+ "-O2",
+ "-Wno-missing-field-initializers",
+ "-Wno-unused-parameter",
+ "-Wno-unused-variable",
+ ],
+ generated_headers: [
+ "tflite_support_spm_config",
+ "tflite_support_spm_encoder_config"
+ ],
+ header_libs: [
+ "tensorflow_headers",
+ "flatbuffer_headers",
+ ],
+ static_libs: [
+ "libtextclassifier_abseil",
+ "libtflite_static",
+ ],
+ apex_available: [
+ "//apex_available:platform",
+ "com.android.extservices",
+ ],
+
+ visibility: ["//external/libtextclassifier:__subpackages__"],
+}
+
+
+// -----------------
+// Generate headers with FlatBuffer schema compiler.
+// -----------------
+genrule_defaults {
+ name: "tflite_support_fbgen",
+ tools: ["flatc"],
+ // "depfile" is used here in conjunction with flatc's -M to gather the deps
+ cmd: "$(location flatc) --cpp --no-union-value-namespacing --gen-object-api --keep-prefix -I external/tflite-support -M $(in) >$(depfile) && " +
+ "$(location flatc) --cpp --no-union-value-namespacing --gen-object-api --keep-prefix -I external/tflite-support -o $$(dirname $(out)) $(in)",
+ depfile: true,
+}
+
+genrule {
+ name: "tflite_support_spm_config",
+ srcs: ["tensorflow_lite_support/custom_ops/kernel/sentencepiece/config.fbs"],
+ out: ["tensorflow_lite_support/custom_ops/kernel/sentencepiece/config_generated.h"],
+ defaults: ["tflite_support_fbgen"],
+}
+
+genrule {
+ name: "tflite_support_spm_encoder_config",
+ srcs: ["tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs"],
+ out: ["tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config_generated.h"],
+ defaults: ["tflite_support_fbgen"],
+}
diff --git a/OWNERS b/OWNERS
new file mode 100644
index 00000000..3cb36c3b
--- /dev/null
+++ b/OWNERS
@@ -0,0 +1,3 @@
+licha@google.com
+toki@google.com
+tonymak@google.com
diff --git a/tensorflow_lite_support/custom_ops/kernel/sentencepiece/decoder_config.fbs b/tensorflow_lite_support/custom_ops/kernel/sentencepiece/decoder_config.fbs
index 4a230ed9..21a155d6 100644
--- a/tensorflow_lite_support/custom_ops/kernel/sentencepiece/decoder_config.fbs
+++ b/tensorflow_lite_support/custom_ops/kernel/sentencepiece/decoder_config.fbs
@@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
-include "config.fbs";
+include "tensorflow_lite_support/custom_ops/kernel/sentencepiece/config.fbs";
namespace tflite.ops.custom.sentencepiece;
diff --git a/tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs b/tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs
index 7f1f2bad..eac962e1 100644
--- a/tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs
+++ b/tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs
@@ -11,7 +11,7 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
-include "config.fbs";
+include "tensorflow_lite_support/custom_ops/kernel/sentencepiece/config.fbs";
namespace tflite.ops.custom.sentencepiece;