Searched refs:region_output_index (Results 1 – 1 of 1) sorted by relevance
401 int region_output_index; in GetTileShardedOutputsToMerge() local404 cluster_func_output_index, ®ion_output_index); in GetTileShardedOutputsToMerge()408 cluster_idx + logical_device_id)[region_output_index]; in GetTileShardedOutputsToMerge()631 int region_output_index; in RemapOutputsFromLogicalDevices() local634 ®ion_output_index))) in RemapOutputsFromLogicalDevices()639 cluster_idx + logical_device_id)[region_output_index]; in RemapOutputsFromLogicalDevices()