Home
last modified time | relevance | path

Searched refs:partial_replication_size (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc560 int64 partial_replication_size = 1; in GatherIndexSharding() local
562 partial_replication_size *= in GatherIndexSharding()
568 Product(index_tile_assignment_dims) * partial_replication_size; in GatherIndexSharding()
571 partial_replication_size *= in GatherIndexSharding()
577 if (partial_replication_size > 1) { in GatherIndexSharding()
578 index_tile_assignment_dims.push_back(partial_replication_size); in GatherIndexSharding()
581 return partial_replication_size > 1 in GatherIndexSharding()