Searched refs:cluster_func_output_index (Results 1 – 1 of 1) sorted by relevance
376 const int cluster_func_output_index) { in MapClusterOutputIndexWithRegionOutputIndex() argument378 for (int output_index = 0; output_index < cluster_func_output_index; in MapClusterOutputIndexWithRegionOutputIndex()392 const int cluster_func_output_index, const xla::OpSharding& sharding, in GetTileShardedOutputsToMerge() argument400 sharding, logical_device_id, cluster_func_output_index); in GetTileShardedOutputsToMerge()411 const int cluster_func_output_index, const xla::OpSharding& sharding, in HandleTileShardedOutputs() argument422 cluster_func_output_index, sharding, parallel_execute); in HandleTileShardedOutputs()