Searched defs:rhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | dot_handler.cc | 437 int64_t rhs_contracting_partitions, int64_t rhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()
768 int64_t rhs_non_contracting_partitions, in PartitionBaseCase()
2010 int64_t rhs_non_contracting_partitions, in PartitionDotGroupOnBatch()
2846 int64_t rhs_non_contracting_partitions, int64_t lhs_matching_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2991 int64_t rhs_non_contracting_partitions, int64_t lhs_contracting_partitions, in PrioritizeContractingDimensionsPartitioning()
3219 int64_t rhs_non_contracting_partitions, int64_t lhs_matching_partitions, in LhsIsBestMatchForNonContractingPartitioning()
3397 const int64_t rhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local
|