Searched refs:dot_dim_numbs (Results 1 – 1 of 1) sorted by relevance
156 const auto& dot_dim_numbs = dot->dot_dimension_numbers(); in ParseDotGeneralFromDot() local158 for (int64 i = 0; i < dot_dim_numbs.lhs_batch_dimensions().size(); ++i) { in ParseDotGeneralFromDot()160 dnums.batch_dims.back().lhs = dot_dim_numbs.lhs_batch_dimensions(i); in ParseDotGeneralFromDot()161 dnums.batch_dims.back().rhs = dot_dim_numbs.rhs_batch_dimensions(i); in ParseDotGeneralFromDot()165 for (int64 i = 0; i < dot_dim_numbs.lhs_contracting_dimensions().size(); in ParseDotGeneralFromDot()169 dot_dim_numbs.lhs_contracting_dimensions(i); in ParseDotGeneralFromDot()171 dot_dim_numbs.rhs_contracting_dimensions(i); in ParseDotGeneralFromDot()176 if (!absl::c_linear_search(dot_dim_numbs.lhs_batch_dimensions(), i) && in ParseDotGeneralFromDot()177 !absl::c_linear_search(dot_dim_numbs.lhs_contracting_dimensions(), i)) { in ParseDotGeneralFromDot()182 dot_dim_numbs.lhs_batch_dimensions_size() + in ParseDotGeneralFromDot()[all …]