diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-12-14 14:48:02 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-12-14 14:55:51 -0800 |
commit | ea703f4e0e72d1e016f8157e206dcc9e80602862 (patch) | |
tree | 092e759fbe5118413efc3cc37f02a008ec9f2711 /tensorflow/contrib/lite/kernels/test_util.cc | |
parent | ab54c4885366e87bd2805036c9c8ab3ffccf823e (diff) |
Implementation of Gather in TfLite
PiperOrigin-RevId: 179101363
Diffstat (limited to 'tensorflow/contrib/lite/kernels/test_util.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/test_util.cc | 13 |
1 files changed, 13 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/test_util.cc b/tensorflow/contrib/lite/kernels/test_util.cc index f716ba8741..b69f2b3e4b 100644 --- a/tensorflow/contrib/lite/kernels/test_util.cc +++ b/tensorflow/contrib/lite/kernels/test_util.cc @@ -180,4 +180,17 @@ int32_t SingleOpModel::GetTensorSize(int index) const { return total_size; } +template <> +std::vector<string> SingleOpModel::ExtractVector(int index) { + TfLiteTensor* tensor_ptr = interpreter_->tensor(index); + CHECK(tensor_ptr != nullptr); + const int num_strings = GetStringCount(tensor_ptr); + std::vector<string> result; + result.reserve(num_strings); + for (int i = 0; i < num_strings; ++i) { + const auto str = GetString(tensor_ptr, i); + result.emplace_back(str.str, str.len); + } + return result; +} } // namespace tflite |