Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h112 HloSharding GatherOutputSharding(const HloSharding& index_sharding,
Dhlo_sharding_util.cc499 HloSharding GatherOutputSharding(const HloSharding& index_sharding, in GatherOutputSharding() function
Dsharding_propagation.cc952 HloSharding new_sharding = hlo_sharding_util::GatherOutputSharding( in InferShardingFromOperands()