Searched refs:GetRegionOutputs (Results 1 – 5 of 5) sorted by relevance
86 auto results = parallel_execute.GetRegionOutputs(i); in SinkResourceWritesIntoParallelExecute()
242 ? parallel_execute.GetRegionOutputs( in BuildVariableAccessInfo()378 auto old_region_results = parallel_execute.GetRegionOutputs(region_index); in ReplaceParallelExecute()
401 const auto output_from_logical_device = parallel_execute.GetRegionOutputs( in GetTileShardedOutputsToMerge()576 parallel_execute.GetRegionOutputs( in RemapOutputsFromLogicalDevices()608 const auto output_from_logical_device = parallel_execute.GetRegionOutputs( in RemapOutputsFromLogicalDevices()
188 Operation::result_range ParallelExecuteOp::GetRegionOutputs( in GetRegionOutputs() function in mlir::tf_device::ParallelExecuteOp
163 Operation::result_range GetRegionOutputs(unsigned region_index);