Home
last modified time | relevance | path

Searched refs:GatherOutputShardingFromDataOperand (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h152 absl::optional<HloSharding> GatherOutputShardingFromDataOperand(
Dhlo_sharding_util.cc946 absl::optional<HloSharding> GatherOutputShardingFromDataOperand( in GatherOutputShardingFromDataOperand() function
Dsharding_propagation.cc974 hlo_sharding_util::GatherOutputShardingFromDataOperand( in InferShardingFromOperands()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc187 hlo_sharding_util::GatherOutputShardingFromDataOperand( in ParititonPassthroughOperand()