Searched defs:lhs_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 381 int64_t rhs_batch_partitions, int64_t lhs_contracting_partitions, in GetWindowedEinsumConfiguration()
520 int64_t output_batch_partitions, int64_t lhs_contracting_partitions, in PartitionBaseCase()
1740 int64_t num_partitions, int64_t lhs_contracting_partitions, in PartitionDotGroupOnBatch()
2487 int64_t rhs_matching_partitions, int64_t lhs_contracting_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2620 int64_t rhs_non_contracting_partitions, int64_t lhs_contracting_partitions, in PrioritizeContractingDimensionsPartitioning()
2768 int64_t rhs_matching_partitions, int64_t lhs_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()
2866 const int64_t lhs_contracting_partitions = in PartitionDot() local
|