Searched refs:rhs_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
1308 absl::InlinedVector<int64, 8> rhs_broadcast_dims = lhs_broadcast_dims; in HandleDotStrengthReduction() local1311 rhs_broadcast_dims.push_back(dot_rank - 1); in HandleDotStrengthReduction()1312 absl::c_sort(rhs_broadcast_dims); in HandleDotStrengthReduction()1318 dot->shape(), rhs_broadcast_dims))))); in HandleDotStrengthReduction()1678 std::vector<int64> rhs_broadcast_dims( in HandleDot() local1680 absl::c_iota(rhs_broadcast_dims, 0); in HandleDot()1682 rhs_broadcast_dims.push_back(i); in HandleDot()1685 dot->shape(), new_rhs, rhs_broadcast_dims)); in HandleDot()1737 std::vector<int64> rhs_broadcast_dims( in HandleDot() local1739 absl::c_iota(rhs_broadcast_dims, 0); in HandleDot()[all …]