Searched defs:output_rhs_non_contracting_partitions (Results 1 – 1 of 1) sorted by relevance
385 int64 output_rhs_non_contracting_partitions, in GetWindowedEinsumConfiguration()479 int64 output_rhs_non_contracting_partitions, in PartitionBaseCase()2080 int64 output_rhs_non_contracting_partitions, const Shape& output_base_shape, in PartitionDotGroupOnContracting()2295 int64 output_rhs_non_contracting_partitions, int64 lhs_batch_partitions, in LhsIsBestMatchForNonContractingPartitioning()2488 const int64 output_rhs_non_contracting_partitions = get_partitions_for_dims( in PartitionDot() local