Searched refs:set_tuple_shapes (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/ |
| H A D | tpu_infeed_test.py | 73 i.set_tuple_shapes([[1], [2, 3]])
75 i.set_tuple_shapes([[1, 2], [3, 4]])
78 i.set_tuple_shapes([[1, 2]])
109 i.set_tuple_shapes([[4, 4], [4]])
115 i.set_tuple_shapes([[8, 8], [8]])
|
| H A D | tpu_feed.py | 199 self.set_tuple_shapes(tuple_shapes)
268 def set_tuple_shapes(self, tuple_shapes): member in InfeedQueue
393 self.set_tuple_shapes([t.shape for t in input_tensors])
438 self.set_tuple_shapes(unsharded_shapes)
|