Searched refs:sharded_inputs (Results 1 – 2 of 2) sorted by relevance
573 sharded_inputs, argument616 self.set_configuration_from_sharded_input_tensors(sharded_inputs)631 for (shard, index) in zip(sharded_inputs, xrange(self.number_of_shards))739 sharded_inputs = [[shard[i] for shard in transposed_sharded_inputs]749 for (shard, index) in zip(sharded_inputs, xrange(self.number_of_shards))817 def generate_enqueue_ops(self, sharded_inputs): argument854 self.set_configuration_from_sharded_input_tensors(sharded_inputs)855 number_of_replicas = len(sharded_inputs)856 number_of_tuple_elements = len(sharded_inputs[0])862 flattened_inputs = sharded_inputs[replica_index][all …]
657 std::vector<NodeOut> sharded_inputs; member3177 sharded_input_info.sharded_inputs.at(core); in BuildExecuteNodes()3261 sharded_input_info.sharded_inputs[core]; in BuildExecuteNodes()