Searched refs:outputs_to_merge (Results 1 – 1 of 1) sorted by relevance
396 llvm::SmallVector<mlir::Value, 4> outputs_to_merge; in GetTileShardedOutputsToMerge() local397 outputs_to_merge.reserve(sharding.tile_assignment_devices_size()); in GetTileShardedOutputsToMerge()403 outputs_to_merge.emplace_back(output_from_logical_device); in GetTileShardedOutputsToMerge()406 return outputs_to_merge; in GetTileShardedOutputsToMerge()421 auto outputs_to_merge = GetTileShardedOutputsToMerge( in HandleTileShardedOutputs() local435 for (int i = 0, end = outputs_to_merge.size(); i < end; in HandleTileShardedOutputs()441 outputs_to_merge.begin() + i, in HandleTileShardedOutputs()442 outputs_to_merge.begin() + i + num_splits}, in HandleTileShardedOutputs()449 std::swap(new_outputs, outputs_to_merge); in HandleTileShardedOutputs()453 assert(outputs_to_merge.size() == 1); in HandleTileShardedOutputs()[all …]