Searched refs:sharded_inputs (Results 1 – 2 of 2) sorted by relevance
548 sharded_inputs, argument591 self.set_configuration_from_sharded_input_tensors(sharded_inputs)606 for (shard, index) in zip(sharded_inputs, xrange(self.number_of_shards))714 sharded_inputs = [[shard[i] for shard in transposed_sharded_inputs]724 for (shard, index) in zip(sharded_inputs, xrange(self.number_of_shards))
619 def __init__(self, sharded_inputs): argument620 self._sharded_inputs = sharded_inputs676 sharded_inputs = self._split_tensors(inputs)677 return self.NumpyInfeedInstance(sharded_inputs)