commit | 1d32b17d97f49eaa250aafc3cf825da4c8098d93 | [log] [tgz] |
---|---|---|
author | Lev Proleev <levp@google.com> | Fri Mar 20 17:44:46 2020 +0000 |
committer | Lev Proleev <levp@google.com> | Mon Apr 06 15:39:28 2020 +0100 |
tree | 1cab31a455f46c3e3cb10cd59b8badace3ab10a9 | |
parent | 83f223a64a83cd8e9aa5a6870daeebb100848f7a [diff] |
Fix HARD_SWISH rank requirement Also add rank check to HARD_SWISH validation. This commit removes nnAssert from the utility function that converts NNAPI Shape to TF Lite's RuntimeShape. The assert would cause a crash if the function was passed a shape with rank > 4 even though RuntimeShape supports such shapes. Bug: 152945527 Bug: 139957496 Test: NNTest_static Change-Id: I69147915669bee525c8883e48fd53ff0f899a501