Searched defs:output_lhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
435 int64_t num_partitions, int64_t output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()769 int64_t output_lhs_non_contracting_partitions, in PartitionBaseCase()2461 int64_t group_count, int64_t output_lhs_non_contracting_partitions, in GetDotGroupPartitionContractingOutputShardings()2580 int64_t output_lhs_non_contracting_partitions, in PartitionDotGroupOnContracting()2849 int64_t output_lhs_non_contracting_partitions, in EstimateWindowedEinsumIterationsForNonContractingPartitioning()2993 int64_t output_lhs_non_contracting_partitions, in PrioritizeContractingDimensionsPartitioning()3222 int64_t output_lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()3399 const int64_t output_lhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local