Searched refs:if_rhs_scalar_builder (Results 1 – 1 of 1) sorted by relevance
263 OpBuilder if_rhs_scalar_builder = in matchAndRewrite() local265 Value reshaped_rhs = if_rhs_scalar_builder.create<mhlo::ReshapeOp>( in matchAndRewrite()267 Value if_rhs_scalar_result = if_rhs_scalar_builder.create<ChloOpTy>( in matchAndRewrite()271 extendToBroadcastShape(if_rhs_scalar_builder, loc, if_rhs_scalar_result, in matchAndRewrite()273 if_rhs_scalar_builder.create<scf::YieldOp>(loc, in matchAndRewrite()