Home
last modified time | relevance | path

Searched refs:old_parallel_execute_output (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dxla_sharding_util.cc563 const auto old_parallel_execute_output = result_and_index.value(); in RemapOutputsFromLogicalDevices() local
573 for (auto user : old_parallel_execute_output.getUsers()) { in RemapOutputsFromLogicalDevices()
581 if (!old_parallel_execute_output.hasOneUse()) in RemapOutputsFromLogicalDevices()
585 << *old_parallel_execute_output.getOwner(); in RemapOutputsFromLogicalDevices()
619 location, old_parallel_execute_output, cluster_idx, in RemapOutputsFromLogicalDevices()
640 old_parallel_execute_output.replaceAllUsesWith(output_from_logical_device); in RemapOutputsFromLogicalDevices()