Searched refs:set_tuple_shapes (Results 1 - 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/tpu/python/tpu/ |
| H A D | tpu_infeed_test.py | 77 i.set_tuple_shapes([[1], [2, 3]])
79 i.set_tuple_shapes([[1, 2], [3, 4]])
82 i.set_tuple_shapes([[1, 2]])
113 i.set_tuple_shapes([[4, 4], [4]])
119 i.set_tuple_shapes([[8, 8], [8]])
|
| H A D | tpu_feed.py | 103 self.set_tuple_shapes(tuple_shapes)
170 def set_tuple_shapes(self, tuple_shapes): member in class:InfeedQueue
292 self.set_tuple_shapes([t.shape for t in input_tensors])
336 self.set_tuple_shapes(unsharded_shapes)
|
Completed in 108 milliseconds