Searched refs:rhs_kept_dim (Results 1 – 1 of 1) sorted by relevance
1166 int64 rhs_kept_dim = kept_dim(rhs_rank, rhs_collapsing_dim, in HandleDotStrengthReduction() local1169 if (rhs_kept_dim == -1 && rhs_rank > 1) { in HandleDotStrengthReduction()1174 std::swap(rhs_collapsing_dim, rhs_kept_dim); in HandleDotStrengthReduction()1261 (rhs_rank == 2 && rhs->shape().dimensions(rhs_kept_dim) == 1)) { in HandleDotStrengthReduction()1299 (rhs_kept_dim == rhs_rank - 1 || in HandleDotStrengthReduction()1300 (rhs_collapsing_dim == rhs_rank - 1 && rhs_kept_dim == rhs_rank - 2))) { in HandleDotStrengthReduction()1325 rhs->shape().dimensions(rhs_kept_dim) == 1 && in HandleDotStrengthReduction()1326 lhs_kept_dim == rhs_kept_dim) { in HandleDotStrengthReduction()1328 auto new_rhs = StripDim(rhs, rhs_kept_dim); in HandleDotStrengthReduction()1329 const int64 reduce_dim = rhs_kept_dim < rhs_collapsing_dim in HandleDotStrengthReduction()[all …]