Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1704 auto output_grouped = GroupShardingOnDims(output_sharding, output_dims); in PartitionDotGroupOnBatch() local
1897 GroupedSharding output_grouped = in GetNonContractingPartitionGroupedShardingForMatchedOperand() local
1927 GroupedSharding output_grouped = in GetNonContractingPartitionGroupedShardingForOtherOperand() local
2008 GroupedSharding output_grouped = in PartitionDotGroupOnNonContracting() local
2371 GroupedSharding output_grouped = in LhsIsBestMatchForNonContractingPartitioning() local