Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1914 int64 matching_contracting_partitions, in GetNonContractingPartitionGroupedShardingForOtherOperand() argument
1941 matching_contracting_partitions == other_contracting_partitions; in GetNonContractingPartitionGroupedShardingForOtherOperand()
1981 int64 matching_contracting_partitions, int64 other_contracting_partitions, in PartitionDotGroupOnNonContracting() argument
2022 matching_contracting_partitions, in PartitionDotGroupOnNonContracting()
2349 const int64 matching_contracting_partitions = in LhsIsBestMatchForNonContractingPartitioning() local
2381 other_non_contracting_partitions, matching_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()
2402 matching_contracting_partitions, 1, matching_batch_partitions, in LhsIsBestMatchForNonContractingPartitioning()