Searched defs:output_grouped (Results 1 – 1 of 1) sorted by relevance
1793 auto output_grouped = GroupShardingOnDims(output_sharding, output_dims); in PartitionDotGroupOnBatch() local1986 GroupedSharding output_grouped = in GetNonContractingPartitionGroupedShardingForMatchedOperand() local2017 GroupedSharding output_grouped = in GetNonContractingPartitionGroupedShardingForOtherOperand() local2099 GroupedSharding output_grouped = in PartitionDotGroupOnNonContracting() local2554 GroupedSharding output_grouped = in EstimateWindowedEinsumIterationsForNonContractingPartitioning() local