Searched refs:cluster_func_output (Results 1 – 1 of 1) sorted by relevance
412 const mlir::Location& location, mlir::Value cluster_func_output, in HandleTileShardedOutputs() argument454 cluster_func_output.replaceAllUsesWith(outputs_to_merge[0]); in HandleTileShardedOutputs()544 const auto cluster_func_output = result_and_index.value(); in RemapOutputsFromLogicalDevices() local554 for (auto user : cluster_func_output.getUsers()) { in RemapOutputsFromLogicalDevices()562 if (!cluster_func_output.hasOneUse()) in RemapOutputsFromLogicalDevices()565 << *cluster_func_output.getOwner(); in RemapOutputsFromLogicalDevices()594 cluster_func_output, parallel_execute, in RemapOutputsFromLogicalDevices()610 cluster_func_output.replaceAllUsesWith(output_from_logical_device); in RemapOutputsFromLogicalDevices()