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()495 int64_t lhs_contracting_dim, int64_t rhs_contracting_dim) { in Dot()
2133 const int64_t rhs_contracting_dim = dnums.rhs_contracting_dimensions(0); in OptimizeDotOfConcat() local2151 HloInstruction* rhs, int64_t rhs_contracting_dim, bool swapped) { in OptimizeDotOfConcatHelper()
2372 int64_t rhs_contracting_dim = dim_numbers.rhs_contracting_dimensions(0); in EmitElementalDot() local