diff options
author | Tony Mak <tonymak@google.com> | 2021-03-12 20:51:46 +0000 |
---|---|---|
committer | Tony Mak <tonymak@google.com> | 2021-03-12 20:51:46 +0000 |
commit | 1101235e137ae68df6530e7fd3639d2cd37e6a62 (patch) | |
tree | 3744c92a1e478183886cde1610bf0dd18fef7889 | |
parent | 3134350bab4acae593e4df99c399c6b357b6bed9 (diff) | |
download | tflite-support-1101235e137ae68df6530e7fd3639d2cd37e6a62.tar.gz |
Make the sentencepiece op and the ragged op compile
This is needed by the new smart reply model.
Test: m tflite-support
Bug: 180325304
Change-Id: Iab778be26a0606ad14b6dba9e1ca14244b1d8611
-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; |