Lines Matching refs:first_shard
638 first_shard = self._inputs[0]640 batch_size = array_ops.shape(first_shard)[0]660 new_center = array_ops.reshape(first_shard[0], [1, -1])671 subset = first_shard[start:end]