Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h128 HloSharding ScatterIndexSharding(const HloSharding& data_sharding,
Dhlo_sharding_util.cc639 HloSharding ScatterIndexSharding(const HloSharding& data_sharding, in ScatterIndexSharding() function
Dsharding_propagation.cc1372 return hlo_sharding_util::ScatterIndexSharding(update->sharding(), in GetShardingFromUser()