Searched refs:rhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 386 int64 rhs_contracting_partitions, int64 rhs_non_contracting_partitions, in GetWindowedEinsumConfiguration() argument
404 if (rhs_non_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration()
477 int64 rhs_non_contracting_partitions, in PartitionBaseCase() argument
1471 rhs_non_contracting_partitions, rhs_batch_partitions, in PartitionBaseCase()
1539 if (rhs_non_contracting_partitions == num_partitions && in PartitionBaseCase()
1653 int64 rhs_non_contracting_partitions, in PartitionDotGroupOnBatch() argument
1856 rhs_contracting_partitions, rhs_non_contracting_partitions, in PartitionDotGroupOnBatch()
2292 int64 rhs_non_contracting_partitions, int64 lhs_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning() argument
2304 rhs_non_contracting_partitions * in LhsIsBestMatchForNonContractingPartitioning()
2345 : rhs_non_contracting_partitions; in LhsIsBestMatchForNonContractingPartitioning()
[all …]
|