Searched refs:rhs_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
397 if (rhs_contracting_partitions == num_partitions) { in GetWindowedEinsumConfiguration()
444 if (lhs_contracting_partitions == rhs_contracting_partitions && in GetWindowedEinsumConfiguration()
476 int64 rhs_contracting_partitions, int64 lhs_non_contracting_partitions, in PartitionBaseCase() argument
1470 output_rhs_non_contracting_partitions, rhs_contracting_partitions, in PartitionBaseCase()
1495 if (lhs_contracting_partitions == rhs_contracting_partitions && in PartitionBaseCase()
1592 return (rhs_contracting_partitions == num_partitions) && in PartitionBaseCase()
1652 int64 rhs_contracting_partitions, int64 lhs_non_contracting_partitions, in PartitionDotGroupOnBatch() argument
1848 rhs_contracting_partitions, in PartitionDotGroupOnBatch()
1856 rhs_contracting_partitions, rhs_non_contracting_partitions, in PartitionDotGroupOnBatch()
[all …]
|