Home
last modified time | relevance | path

Searched defs:output_grouped (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dgather_scatter_handler.cc200 GroupedSharding output_grouped = hlo_sharding_util::GroupShardingOnDims( in PartitionGatherIndexPassthroughPartition() local
307 GroupedSharding output_grouped = hlo_sharding_util::GroupShardingOnDims( in PartitionGatherPassthroughOperand() local
H A Ddot_handler.cc2063 auto output_grouped = in PartitionDotGroupOnBatch() local
2260 GroupedSharding output_grouped = in GetNonContractingPartitionGroupedShardingForMatchedOperand() local
2292 GroupedSharding output_grouped = in GetNonContractingPartitionGroupedShardingForOtherOperand() local
2377 GroupedSharding output_grouped = in PartitionDotGroupOnNonContracting() local
2914 GroupedSharding output_grouped = in EstimateWindowedEinsumIterationsForNonContractingPartitioning() local