Searched defs:rhs_contracting_dim (Results 1 – 3 of 3) sorted by relevance
156 int64 rhs_contracting_dim) in HloDotWithContractingDimsMatcher()407 int64 lhs_contracting_dim, int64 rhs_contracting_dim) { in Dot()
1372 const int64 rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat() local1390 HloInstruction* rhs, int64 rhs_contracting_dim, bool swapped) { in OptimizeDotOfConcatHelper()
2120 int64 rhs_contracting_dim = dim_numbers.rhs_contracting_dimensions(0); in EmitElementalDot() local