Searched defs:lhs_contracting_dim (Results 1 – 3 of 3) sorted by relevance
155 ::testing::Matcher<const HloInstruction*> rhs, int64 lhs_contracting_dim, in HloDotWithContractingDimsMatcher()407 int64 lhs_contracting_dim, int64 rhs_contracting_dim) { in Dot()
1371 const int64 lhs_contracting_dim = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfConcat() local1389 const HloInstruction& dot, HloInstruction* lhs, int64 lhs_contracting_dim, in OptimizeDotOfConcatHelper()
2119 int64 lhs_contracting_dim = dim_numbers.lhs_contracting_dimensions(0); in EmitElementalDot() local