diff options
| author | Miao Wang <miaowang@google.com> | 2019-03-15 15:46:12 -0700 |
|---|---|---|
| committer | Miao Wang <miaowang@google.com> | 2019-03-16 00:08:45 +0000 |
| commit | 2698f1c2d4403119f7dc12677476235d24fda6db (patch) | |
| tree | 16a8ad9cf81aa732f3fe2f17e82f7e8221a41ad5 | |
| parent | fbc233e169ed87603a0fc89062de923d5b59f633 (diff) | |
| download | libtextclassifier-2698f1c2d4403119f7dc12677476235d24fda6db.tar.gz | |
Use WriteToTensorAsVector instead of WriteToTensor
Test: mm
Change-Id: If43f1caf5e1cce6caba065e1319b934737bfffba
Merged-In: I3928896f37640a3923cc7c0ba206c643506fbcbd
| -rw-r--r-- | utils/tflite-model-executor.cc | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/utils/tflite-model-executor.cc b/utils/tflite-model-executor.cc index 4eb0449..e3ec73c 100644 --- a/utils/tflite-model-executor.cc +++ b/utils/tflite-model-executor.cc @@ -97,9 +97,8 @@ void TfLiteModelExecutor::SetInput(const int input_index, for (const std::string& s : input_data) { buf.AddString(s.data(), s.length()); } - // TODO(b/120230709): Use WriteToTensorAsVector() instead, once available in - // AOSP. - buf.WriteToTensor(interpreter->tensor(interpreter->inputs()[input_index])); + buf.WriteToTensorAsVector( + interpreter->tensor(interpreter->inputs()[input_index])); } template <> |