Searched refs:outputs_to_merge (Results 1 – 1 of 1) sorted by relevance
394 llvm::SmallVector<mlir::Value, 4>* outputs_to_merge) { in GetTileShardedOutputsToMerge() argument399 outputs_to_merge->reserve(sharding.tile_assignment_devices_size()); in GetTileShardedOutputsToMerge()409 outputs_to_merge->emplace_back(output_from_logical_device); in GetTileShardedOutputsToMerge()429 llvm::SmallVector<mlir::Value, 4> outputs_to_merge; in HandleTileShardedOutputs() local433 &outputs_to_merge); in HandleTileShardedOutputs()449 for (int i = 0, end = outputs_to_merge.size(); i < end; in HandleTileShardedOutputs()454 outputs_to_merge.begin() + i, in HandleTileShardedOutputs()455 outputs_to_merge.begin() + i + num_splits}, in HandleTileShardedOutputs()460 std::swap(new_outputs, outputs_to_merge); in HandleTileShardedOutputs()464 assert(outputs_to_merge.size() == 1); in HandleTileShardedOutputs()[all …]