experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add a helper error message when the tensor index is not found.

Browse Source 
PiperOrigin-RevId: 316155846
Change-Id: I0754d7520e9ee3ec352fd26691581ac05f2caa3e
This commit is contained in:
Gaurav Menghani 2020-06-12 12:33:14 -07:00 committed by TensorFlower Gardener
parent 0511d9fd95
commit 204f6a1c5d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/lite/kernels/test_util.h
View File

@ -461,7 +461,7 @@ class SingleOpModel {
std::vector<T> ExtractVector(int index) const { std::vector<T> ExtractVector(int index) const {
const T* v = interpreter_->typed_tensor<T>(index); const T* v = interpreter_->typed_tensor<T>(index);
const auto* tensor = interpreter_->tensor(index); const auto* tensor = interpreter_->tensor(index);
CHECK(v); CHECK(v) << "Could not extract vector at index: " << index;
int tensor_size; int tensor_size;
if (tensor->sparsity) { if (tensor->sparsity) {
// Getting the size of the sparse buffer this way is based on the // Getting the size of the sparse buffer this way is based on the