Searched refs:_compareSqueezeAll (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | shape_ops_test.py | 289 def _compareSqueezeAll(self, x, squeeze_dims=None): member in ShapeOpsTest
297 self._compareSqueezeAll(np.zeros([2]))
298 self._compareSqueezeAll(np.zeros([2, 3]))
301 self._compareSqueezeAll(np.zeros([2, 1, 2]))
304 self._compareSqueezeAll(np.zeros([1, 2, 1, 3, 1]))
309 self._compareSqueezeAll(choice([2]))
310 self._compareSqueezeAll(choice([2, 3]))
313 self._compareSqueezeAll(choice([2, 1, 2]))
316 self._compareSqueezeAll(choice([1, 2, 1, 3, 1]))
320 self._compareSqueezeAll(np.zeros([1, 2, 1, 3, 1]), [0])
[all …]
|