Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h133 HloSharding ScatterDataSharding(const HloSharding& index_sharding,
Dhlo_sharding_util.cc672 HloSharding ScatterDataSharding(const HloSharding& index_sharding, in ScatterDataSharding() function
Dsharding_propagation.cc1379 hlo_sharding_util::ScatterDataSharding(indices->sharding(), &user); in GetShardingFromUser()