Searched refs:_flat_shapes (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 153 def _flat_shapes(dataset): function
172 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
182 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
191 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
200 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
209 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
218 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
232 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
245 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
257 self.assertEqual(expected_shapes, _flat_shapes(rebatched_dataset))
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | readers.py | 798 output_shapes=self._flat_shapes,
811 output_shapes=self._flat_shapes,
|
| /external/tensorflow/tensorflow/python/data/ops/ |
| D | dataset_ops.py | 539 def _flat_shapes(self): member in DatasetV2
571 "output_shapes": self._flat_shapes,
4809 flat_shapes = input_dataset._flat_shapes # pylint: disable=protected-access
|