Searched refs:num_towers (Results 1 – 1 of 1) sorted by relevance
265 def num_towers(self): member in TPUAssignment659 if self._tpu_assignment.num_towers == 1:663 assert batch_size % self._tpu_assignment.num_towers == 0, (665 'vs %s)' % (batch_size, self._tpu_assignment.num_towers))666 shard_size = batch_size // self._tpu_assignment.num_towers668 for index in range(self._tpu_assignment.num_towers):683 for shard_id in range(self._tpu_assignment.num_towers):741 dummy_x_shape[0] *= tpu_assignment.num_towers743 dummy_y_shape[0] *= tpu_assignment.num_towers749 for i in range(tpu_assignment.num_towers):[all …]