Home
last modified time | relevance | path

Searched refs:TransposeShardingWithCollapsedDims (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.h201 absl::optional<HloSharding> TransposeShardingWithCollapsedDims(
Dsharding_propagation.cc378 return *hlo_sharding_util::TransposeShardingWithCollapsedDims( in InferDotShardingFromOperands()
1089 auto sharding = *hlo_sharding_util::TransposeShardingWithCollapsedDims( in InferDotOperandSharding()
1111 *hlo_sharding_util::TransposeShardingWithCollapsedDims( in InferDotOperandSharding()
Dhlo_sharding_util.cc1206 absl::optional<HloSharding> TransposeShardingWithCollapsedDims( in TransposeShardingWithCollapsedDims() function
1218 return TransposeShardingWithCollapsedDims(source, new_src_to_tgt, in TransposeShardingWithCollapsedDims()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc495 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionBaseCase()
499 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionBaseCase()
503 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionBaseCase()
507 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionBaseCase()
511 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionBaseCase()
515 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionBaseCase()
2386 hlo_sharding_util::TransposeShardingWithCollapsedDims( in LhsIsBestMatchForNonContractingPartitioning()
2390 hlo_sharding_util::TransposeShardingWithCollapsedDims( in LhsIsBestMatchForNonContractingPartitioning()
Dgather_scatter_handler.cc162 hlo_sharding_util::TransposeShardingWithCollapsedDims( in PartitionIndexOnlyPartition()
573 hlo_sharding_util::TransposeShardingWithCollapsedDims( in HandleScatter()