Searched defs:cluster_func_output_index (Results 1 – 1 of 1) sorted by relevance
376 const int cluster_func_output_index) { in MapClusterOutputIndexWithRegionOutputIndex()392 const int cluster_func_output_index, const xla::OpSharding& sharding, in GetTileShardedOutputsToMerge()411 const int cluster_func_output_index, const xla::OpSharding& sharding, in HandleTileShardedOutputs()