Searched refs:else_builder (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | transform_unranked_hlo.cc | 439 OpBuilder else_builder = if_op.getElseBodyBuilder(rewriter.getListener()); in HandleBroadcastAndOp() local
443 else_builder, op, greater_rank, i); in HandleBroadcastAndOp()
446 else_builder.create<scf::YieldOp>(loc, inner_if.getResult(0)); in HandleBroadcastAndOp()
447 else_builder = inner_if.getElseBodyBuilder(rewriter.getListener()); in HandleBroadcastAndOp()
451 else_builder.create<AssertOp>( in HandleBroadcastAndOp()
453 GreaterRankIsN(else_builder, op.getLoc(), greater_rank, in HandleBroadcastAndOp()
458 createRankSpecializedBroadcastAndOp(else_builder, op, lhs, rhs, in HandleBroadcastAndOp()
|
| D | lhlo_legalize_to_parallel_loops.cc | 446 OpBuilder else_builder = in CreateReduceOpInNestedParallelLoops() local
448 else_builder.create<scf::YieldOp>(loc, *window_loop.initVals().begin()); in CreateReduceOpInNestedParallelLoops()
|