Home
last modified time | relevance | path

Searched defs:rhs_sharding (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc414 const HloSharding& lhs_sharding, const HloSharding& rhs_sharding, in RequiresTransposeSharding()
446 const HloSharding& lhs_sharding, const HloSharding& rhs_sharding, in GetWindowedEinsumConfiguration()
777 const HloSharding& rhs_sharding = rhs.sharding(); in PartitionBaseCase() local
2547 HloSharding rhs_sharding = rhs.sharding(); in GetDotGroupPartitionContractingLhsRhsShardings() local
2609 HloSharding rhs_sharding = HloSharding::Replicate(); in PartitionDotGroupOnContracting() local
3051 HloSharding rhs_sharding = HloSharding::Replicate(); in PrioritizeContractingDimensionsPartitioning() local
H A Dconvolution_handler.cc259 const HloSharding& rhs_sharding) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
572 const HloSharding& rhs_sharding) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_sharding_util_test.cc440 HloSharding rhs_sharding = in TEST() local