Searched refs:rhs_contracting_dimensions_size (Results 1 – 8 of 8) sorted by relevance
117 CHECK_EQ(new_dim_numbers.rhs_contracting_dimensions_size(), 1); in FoldTransposeIntoDot()
235 if (dim_nums.rhs_contracting_dimensions_size() != 1 || in MatchAndExplain()
1962 dnums.rhs_contracting_dimensions_size() != 1 || in OptimizeDotOfGather()2374 (dot->dot_dimension_numbers().rhs_contracting_dimensions_size() + in HandleDot()2406 dot->dot_dimension_numbers().rhs_contracting_dimensions_size()); in HandleDot()5100 if (rhs->shape().rank() != 1 + dnums.rhs_contracting_dimensions_size()) { in HandleTranspose()
1323 if (dot->dot_dimension_numbers().rhs_contracting_dimensions_size() == 1 &&1418 (rhs_rank - dnums.rhs_contracting_dimensions_size()));
670 dimension_numbers.rhs_contracting_dimensions_size()) { in InferDotOpShape()
188 dimension_numbers.rhs_contracting_dimensions_size()) in __anon3500048a0202()193 dimension_numbers.rhs_contracting_dimensions_size()); in __anon3500048a0202()
274 CHECK_EQ(dot_info_.dim_nums.rhs_contracting_dimensions_size(), 1); in EmitLinalgMatmul()863 CHECK_GE(dim_nums.rhs_contracting_dimensions_size(), 0); in GetMatMultDims()
183 int64_t rhs_contracting_dimensions_size = llvm::size( in Verify() local185 if (lhs_contracting_dimensions_size != rhs_contracting_dimensions_size) { in Verify()