Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc2243 GroupedSharding GetNonContractingPartitionGroupedShardingForMatchedOperand( in GetNonContractingPartitionGroupedShardingForMatchedOperand() function
2380 GetNonContractingPartitionGroupedShardingForMatchedOperand( in PartitionDotGroupOnNonContracting()
2917 GetNonContractingPartitionGroupedShardingForMatchedOperand( in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
3275 GetNonContractingPartitionGroupedShardingForMatchedOperand( in LhsIsBestMatchForNonContractingPartitioning()
3280 GetNonContractingPartitionGroupedShardingForMatchedOperand( in LhsIsBestMatchForNonContractingPartitioning()