Searched refs:original_dot (Results 1 – 1 of 1) sorted by relevance
165 Status CanonicalizeDot(HloInstruction* original_dot) { in CanonicalizeDot() argument166 auto computation = original_dot->parent(); in CanonicalizeDot()167 const auto& original_dnums = original_dot->dot_dimension_numbers(); in CanonicalizeDot()172 const auto& lhs_shape = original_dot->operand(0)->shape(); in CanonicalizeDot()210 original_dot->mutable_operand(0), lhs_transpose)); in CanonicalizeDot()220 const auto& rhs_shape = original_dot->operand(1)->shape(); in CanonicalizeDot()254 original_dot->mutable_operand(1), rhs_transpose)); in CanonicalizeDot()278 ShapeUtil::MakeShape(original_dot->shape().element_type(), dot_dims), in CanonicalizeDot()279 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config())); in CanonicalizeDot()282 original_dot, computation->AddInstruction(HloInstruction::CreateReshape( in CanonicalizeDot()[all …]