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