Searched refs:original_dnums (Results 1 – 1 of 1) sorted by relevance
167 const auto& original_dnums = original_dot->dot_dimension_numbers(); in CanonicalizeDot() local168 const int64 num_batch_dims = original_dnums.lhs_batch_dimensions_size(); in CanonicalizeDot()170 original_dnums.lhs_contracting_dimensions_size(); in CanonicalizeDot()184 if (absl::c_linear_search(original_dnums.lhs_contracting_dimensions(), i)) { in CanonicalizeDot()186 } else if (absl::c_linear_search(original_dnums.lhs_batch_dimensions(), in CanonicalizeDot()199 original_dnums.lhs_batch_dimensions().begin(), in CanonicalizeDot()200 original_dnums.lhs_batch_dimensions().end()); in CanonicalizeDot()204 original_dnums.lhs_contracting_dimensions().begin(), in CanonicalizeDot()205 original_dnums.lhs_contracting_dimensions().end()); in CanonicalizeDot()229 if (absl::c_linear_search(original_dnums.rhs_contracting_dimensions(), i)) { in CanonicalizeDot()[all …]