Searched refs:dot_rank (Results 1 – 1 of 1) sorted by relevance
1143 const int64 dot_rank = dot->shape().rank(); in HandleDotStrengthReduction() local1150 if (dot_rank > 2 && (lhs_rank != rhs_rank || lhs_rank != dot_rank)) { in HandleDotStrengthReduction()1271 if (dot_rank <= 2) { in HandleDotStrengthReduction()1276 CHECK_EQ(dot_rank, lhs_rank); in HandleDotStrengthReduction()1280 if (dnums.rhs_batch_dimensions_size() + 2 != dot_rank || in HandleDotStrengthReduction()1311 rhs_broadcast_dims.push_back(dot_rank - 1); in HandleDotStrengthReduction()