Searched refs:reshaped_lhs (Results 1 – 2 of 2) sorted by relevance
91 HloInstruction* reshaped_lhs = in CanonicalizeDot() local162 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config())); in CanonicalizeDot()
242 Value reshaped_lhs = if_lhs_scalar_builder.create<mhlo::ReshapeOp>( in matchAndRewrite() local245 loc, ArrayRef<Type>{result_type}, ArrayRef<Value>{reshaped_lhs, rhs}, in matchAndRewrite()390 Value reshaped_lhs = if_builder.create<mhlo::DynamicReshapeOp>( in createRankSpecializedBroadcastAndOp() local402 ArrayRef<Value>{reshaped_lhs, reshaped_rhs}, op.getAttrs()); in createRankSpecializedBroadcastAndOp()