Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h158 absl::optional<HloSharding> GatherDataOperandShardingFromOutput(
Dhlo_sharding_util.cc961 absl::optional<HloSharding> GatherDataOperandShardingFromOutput( in GatherDataOperandShardingFromOutput() function
Dsharding_propagation.cc1358 return hlo_sharding_util::GatherDataOperandShardingFromOutput( in GetShardingFromUser()