Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h117 HloSharding GatherIndexSharding(const HloSharding& output_sharding,
Dhlo_sharding_util.cc537 HloSharding GatherIndexSharding(const HloSharding& output_sharding, in GatherIndexSharding() function
Dsharding_propagation.cc1355 return hlo_sharding_util::GatherIndexSharding(user.sharding(), &user); in GetShardingFromUser()