Searched defs:lhs_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()
1886 const int64_t lhs_contracting_dim = dnums.lhs_contracting_dimensions(0); in OptimizeDotOfConcat() local1904 const HloInstruction& dot, HloInstruction* lhs, int64_t lhs_contracting_dim, in OptimizeDotOfConcatHelper()
2364 int64 lhs_contracting_dim = dim_numbers.lhs_contracting_dimensions(0); in EmitElementalDot() local