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

Add more check to sparsity parameter verifier.

Browse Source 
PiperOrigin-RevId: 286436553
Change-Id: I43913b2a16ad7bb3b3e22fa65cd47462159b8f67
This commit is contained in:
Yunlu Li 2019-12-19 12:00:12 -08:00 committed by TensorFlower Gardener
parent 0c43ecf472
commit ba91c04e00
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/tools/verifier.cc
View File

@ -179,7 +179,8 @@ absl::optional<uint64_t> VerifyAndCountSparseElements(const Tensor& tensor) {
const int total_dims = sparsity->traversal_order()->size();
if (sparsity->dim_metadata()->size() != total_dims) {
if (total_dims < tensor.shape()->size() ||
sparsity->dim_metadata()->size() != total_dims) {
return absl::nullopt;
}