commit | e93f42e33491c962f1d1dfaca8696618f750d997 | [log] [tgz] |
---|---|---|
author | Przemysław Szczepaniak <pszczepaniak@google.com> | Wed Apr 29 19:47:05 2020 +0100 |
committer | Xusong Wang <xusongw@google.com> | Fri May 01 15:42:58 2020 -0700 |
tree | 8585f7292a59887d9f1600ba4da67afac158dc05 | |
parent | a733bd722f448b5d0586f9422f5f953b734fd832 [diff] |
Fix null ptr dereference in squeeze cpu implementation. Squeeze, when used on a single-value tensor with 1s in each dimension, will result in zero-sized tensor without dimensions. This change fixes it, by making its output a tensor of size [1]. Bug: 155238914 Test: squeeze_b155238914.mod.py run on coral. Change-Id: Ia5bf655c647084d9fa094ae8a2048917a8237b81 Merged-In: Ia5bf655c647084d9fa094ae8a2048917a8237b81 (cherry picked from commit 65c39cb47134728db1f72231b34a9472081f9840)