diff options
author | fban <fban@google.com> | 2022-09-09 19:49:04 +0000 |
---|---|---|
committer | Presubmit Automerger Backend <android-build-presubmit-automerger-backend@system.gserviceaccount.com> | 2022-09-09 19:49:04 +0000 |
commit | 1aba09d77f566289d66b009f449ea0be9eb95639 (patch) | |
tree | 659fa77eb74c1af2bc08d8f2101fc77872fac0a9 | |
parent | c3d19e7b8cb9e8bd202fd460c053bddd0c48670d (diff) | |
parent | abaa5ec61b7e9ee979f9c2f1f596d4fa85bbabeb (diff) | |
download | tflite-support-1aba09d77f566289d66b009f449ea0be9eb95639.tar.gz |
[automerge] Adds a sanity check. 2p: abaa5ec61b
Original change: https://googleplex-android-review.googlesource.com/c/platform/external/tflite-support/+/19586649
Bug: 241507692
Change-Id: Ia0c4122c012023fc7a77fbb45e821312bfb9e499
-rw-r--r-- | tensorflow_lite_support/cc/task/text/nlclassifier/bert_nl_classifier.cc | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/tensorflow_lite_support/cc/task/text/nlclassifier/bert_nl_classifier.cc b/tensorflow_lite_support/cc/task/text/nlclassifier/bert_nl_classifier.cc index d403755b..88a8a0af 100644 --- a/tensorflow_lite_support/cc/task/text/nlclassifier/bert_nl_classifier.cc +++ b/tensorflow_lite_support/cc/task/text/nlclassifier/bert_nl_classifier.cc @@ -246,6 +246,14 @@ absl::Status BertNLClassifier::InitializeFromMetadata() { TfLiteSupportStatus::kInvalidInputTensorSizeError); } + // If some tensor does not have a size 2 dims_signature, then we + // assume the input is not dynamic. + if (ids_tensor.dims_signature->size != 2 || + mask_tensor.dims_signature->size != 2 || + segment_ids_tensor.dims_signature->size != 2) { + return absl::OkStatus(); + } + if (ids_tensor.dims_signature->data[1] == -1 && mask_tensor.dims_signature->data[1] == -1 && segment_ids_tensor.dims_signature->data[1] == -1) { |