Searched refs:lhs_broadcast_dims (Results 1 – 1 of 1) sorted by relevance
1304 absl::InlinedVector<int64, 8> lhs_broadcast_dims; in HandleDotStrengthReduction() local1306 lhs_broadcast_dims.push_back(dim > lhs_collapsing_dim ? (dim - 1) : dim); in HandleDotStrengthReduction()1308 absl::InlinedVector<int64, 8> rhs_broadcast_dims = lhs_broadcast_dims; in HandleDotStrengthReduction()1309 lhs_broadcast_dims.push_back(lhs_kept_dim_in_output); in HandleDotStrengthReduction()1310 absl::c_sort(lhs_broadcast_dims); in HandleDotStrengthReduction()1316 dot->shape(), lhs_broadcast_dims), in HandleDotStrengthReduction()1665 std::vector<int64> lhs_broadcast_dims(lhs->shape().rank()); in HandleDot() local1666 absl::c_iota(lhs_broadcast_dims, 0); in HandleDot()1668 dot->shape(), new_lhs, lhs_broadcast_dims)); in HandleDot()1725 std::vector<int64> lhs_broadcast_dims( in HandleDot() local[all …]