Searched refs:_open_ring_2d (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | device_assignment.py | 187 def _open_ring_2d(x_size: int, y_size: int, function
302 ret.extend(_open_ring_2d(x_size, y_size, 0))
306 ret.extend((x, y, z) for (x, z, y) in _open_ring_2d(x_size, z_size, 0))
310 ret.extend((x, y, z) for (y, z, x) in _open_ring_2d(y_size, z_size, 0))
316 r = _open_ring_2d(x_size, y_size, i)
|