Searched refs:GetRegionOutputs (Results 1 – 5 of 5) sorted by relevance
407 new_parallel_execute.GetRegionOutputs( in GetTileShardedOutputsToMerge()596 new_parallel_execute.GetRegionOutputs( in RemapOutputsFromLogicalDevices()638 new_parallel_execute.GetRegionOutputs( in RemapOutputsFromLogicalDevices()
89 auto results = parallel_execute.GetRegionOutputs(i); in SinkResourceWritesIntoParallelExecute()
262 ? parallel_execute.GetRegionOutputs( in BuildVariableAccessInfo()407 auto old_region_results = parallel_execute.GetRegionOutputs(region_index); in ReplaceParallelExecute()
192 Operation::result_range ParallelExecuteOp::GetRegionOutputs( in GetRegionOutputs() function in mlir::tf_device::ParallelExecuteOp
165 Operation::result_range GetRegionOutputs(unsigned region_index);