diff options
author | Miao Wang <miaowang@google.com> | 2019-03-18 09:25:47 -0700 |
---|---|---|
committer | android-build-merger <android-build-merger@google.com> | 2019-03-18 09:25:47 -0700 |
commit | fec6874c1fb9e3f5d58dc562dc2e942f0ded2bab (patch) | |
tree | 1fc7c6eac4ef33766ad799666162f29082f06bfa | |
parent | 88a3b0161360589d597e5fb9a69426d0d282f95d (diff) | |
parent | 6ce16c28e5434513386c1dac89661ef7deac48af (diff) | |
download | libtextclassifier-fec6874c1fb9e3f5d58dc562dc2e942f0ded2bab.tar.gz |
Merge "Use WriteToTensorAsVector instead of WriteToTensor" am: 87db7aab32
am: 6ce16c28e5
Change-Id: I087fc1d7ca695c90ba73843c4300c7bde35e2013
-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 <> |