Searched refs:_getArrayShapes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert_saved_model_test.py | 148 def _getArrayShapes(self, tensors): member in FreezeSavedModelTest
187 self.assertEqual(self._getArrayShapes(in_tensors), [[1, 16, 16, 3]])
196 self.assertEqual(self._getArrayShapes(in_tensors), [[None, 16, 16, 3]])
242 self.assertEqual(self._getArrayShapes(in_tensors), [[None, 16, 16, 3]])
254 self.assertEqual(self._getArrayShapes(in_tensors), [[1, 16, 16, 3]])
266 self._getArrayShapes(in_tensors), [[1, 16, 16, 3], [1, 16, 16, 3]])
280 self.assertEqual(self._getArrayShapes(in_tensors), [[1, 16, 16, 3]])
288 self.assertEqual(self._getArrayShapes(in_tensors), [[1, 16, 16, 3]])
326 self.assertEqual(self._getArrayShapes(in_tensors), [[1, 28, 28]])
|