diff options
author | Tony Mak <tonymak@google.com> | 2021-03-15 15:46:58 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-03-15 15:46:58 +0000 |
commit | cd41010b1c4a65b1514890aaab5bce96b857007b (patch) | |
tree | 3744c92a1e478183886cde1610bf0dd18fef7889 | |
parent | 9e521358599847a7ffe6fd108dfa3a3c2b7d31e2 (diff) | |
parent | c07ecc9d3f616cdeffe7c815b1ba2f8df46ec6a7 (diff) | |
download | tflite-support-cd41010b1c4a65b1514890aaab5bce96b857007b.tar.gz |
Make the sentencepiece op and the ragged op compile am: 1101235e13 am: c07ecc9d3f
Original change: https://android-review.googlesource.com/c/platform/external/tflite-support/+/1634620
MUST ONLY BE SUBMITTED BY AUTOMERGER
Change-Id: Ibdb343c3cd3b5b5ad19f0132f0ae6db4d8d350ba
-rw-r--r-- | Android.bp | 36 | ||||
-rw-r--r-- | tensorflow_lite_support/custom_ops/kernel/sentencepiece/decoder_config.fbs | 2 | ||||
-rw-r--r-- | tensorflow_lite_support/custom_ops/kernel/sentencepiece/encoder_config.fbs | 2 |
3 files changed, 38 insertions, 2 deletions
@@ -6,6 +6,10 @@ cc_library_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", @@ -15,8 +19,13 @@ cc_library_static { "-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", @@ -29,3 +38,30 @@ cc_library_static { 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/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; |