Searched refs:lhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 388 int64 lhs_non_contracting_partitions, int64 lhs_batch_partitions, in GetWindowedEinsumConfiguration() argument
429 if (lhs_non_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration()
476 int64 rhs_contracting_partitions, int64 lhs_non_contracting_partitions, in PartitionBaseCase() argument
1472 lhs_contracting_partitions, lhs_non_contracting_partitions, in PartitionBaseCase()
1529 if (lhs_non_contracting_partitions == num_partitions && in PartitionBaseCase()
1652 int64 rhs_contracting_partitions, int64 lhs_non_contracting_partitions, in PartitionDotGroupOnBatch() argument
1847 lhs_contracting_partitions, lhs_non_contracting_partitions, in PartitionDotGroupOnBatch()
2291 int64 num_partitions, int64 lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning() argument
2302 lhs_non_contracting_partitions * in LhsIsBestMatchForNonContractingPartitioning()
2344 assume_lhs_match ? lhs_non_contracting_partitions in LhsIsBestMatchForNonContractingPartitioning()
[all …]
|