Searched refs:kept_dim (Results 1 – 1 of 1) sorted by relevance
1132 const auto kept_dim = [](int64 rank, int64 contracting_dimension, in HandleDotStrengthReduction() local1154 int64 lhs_kept_dim = kept_dim(lhs_rank, lhs_collapsing_dim, in HandleDotStrengthReduction()1166 int64 rhs_kept_dim = kept_dim(rhs_rank, rhs_collapsing_dim, in HandleDotStrengthReduction()