Searched refs:reshaped_rhs (Results 1 – 2 of 2) sorted by relevance
138 HloInstruction* reshaped_rhs = in CanonicalizeDot() local162 reshaped_lhs, reshaped_rhs, dot_dnums, original_dot->precision_config())); in CanonicalizeDot()
265 Value reshaped_rhs = if_rhs_scalar_builder.create<mhlo::ReshapeOp>( in matchAndRewrite() local268 loc, ArrayRef<Type>{result_type}, ArrayRef<Value>{lhs, reshaped_rhs}, in matchAndRewrite()392 Value reshaped_rhs = if_builder.create<mhlo::DynamicReshapeOp>( in createRankSpecializedBroadcastAndOp() local402 ArrayRef<Value>{reshaped_lhs, reshaped_rhs}, op.getAttrs()); in createRankSpecializedBroadcastAndOp()