Searched defs:output_lhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
384 int64 num_partitions, int64 output_lhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()478 int64 output_lhs_non_contracting_partitions, in PartitionBaseCase()2079 int64 output_batch_partitions, int64 output_lhs_non_contracting_partitions, in PartitionDotGroupOnContracting()2294 int64 output_lhs_non_contracting_partitions, in LhsIsBestMatchForNonContractingPartitioning()2486 const int64 output_lhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local