Searched defs:lhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | dot_handler.cc | 439 int64_t lhs_non_contracting_partitions, int64_t lhs_batch_partitions, in GetWindowedEinsumConfiguration()
767 int64_t rhs_contracting_partitions, int64_t lhs_non_contracting_partitions, in PartitionBaseCase()
2009 int64_t rhs_contracting_partitions, int64_t lhs_non_contracting_partitions, in PartitionDotGroupOnBatch()
2845 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()
2990 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in PrioritizeContractingDimensionsPartitioning()
3218 int64_t num_partitions, int64_t lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()
3395 const int64_t lhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local
|