Searched defs:rhs_contracting_dim (Results 1 – 3 of 3) sorted by relevance
162 int64_t lhs_contracting_dim, int64_t rhs_contracting_dim) in HloDotWithContractingDimsMatcher()477 int64_t lhs_contracting_dim, int64_t rhs_contracting_dim) { in Dot()
1887 const int64_t rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat() local1905 HloInstruction* rhs, int64_t rhs_contracting_dim, bool swapped) { in OptimizeDotOfConcatHelper()
2365 int64 rhs_contracting_dim = dim_numbers.rhs_contracting_dimensions(0); in EmitElementalDot() local