Home
last modified time | relevance | path

Searched refs:region_if (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_communication.cc547 void RewriteRegionIfOp(OpBuilder& builder, IfOp region_if, in RewriteRegionIfOp() argument
553 Value new_true_operand = GetValueWithToken(builder, region_if.true_arg(), in RewriteRegionIfOp()
555 Value new_false_operand = GetValueWithToken(builder, region_if.false_arg(), in RewriteRegionIfOp()
558 auto new_result_type = GetTypeWithToken(builder, region_if.getType()); in RewriteRegionIfOp()
561 auto new_if = builder.create<IfOp>(region_if.getLoc(), new_result_type, in RewriteRegionIfOp()
562 region_if.pred(), new_true_operand, in RewriteRegionIfOp()
566 new_if.true_branch().takeBody(region_if.true_branch()); in RewriteRegionIfOp()
567 new_if.false_branch().takeBody(region_if.false_branch()); in RewriteRegionIfOp()
571 ReplaceWithTupleResult(builder, region_if.getResult(), new_if.getResult()); in RewriteRegionIfOp()
577 region_if.erase(); in RewriteRegionIfOp()
[all …]
/external/llvm-project/mlir/test/Transforms/
Dbuffer-hoisting.mlir468 %1 = test.region_if %0 : memref<?x?xf32> -> (memref<?x?xf32>) then {
482 // CHECK-NEXT: {{.*}} test.region_if
495 %1 = test.region_if %0 : memref<?x?xf32> -> (memref<?x?xf32>) then {
511 // CHECK-NEXT: {{.*}} test.region_if
Dpromote-buffers-to-stack.mlir447 %1 = test.region_if %0 : memref<2x2xf32> -> (memref<2x2xf32>) then {
461 // CHECK-NEXT: %[[ALLOC1:.*]] = test.region_if
Dbuffer-deallocation.mlir664 %1 = test.region_if %0 : memref<?x?xf32> -> (memref<?x?xf32>) then {
678 // CHECK-NEXT: %[[ALLOC1:.*]] = test.region_if
/external/llvm-project/mlir/docs/
DBufferization.md784 “custom.region_if” operation. It either executes the “then” or “else” region
795 %1 = custom.region_if %0 : memref<?x?xf32> -> (memref<?x?xf32>)
856 Reconsider a slightly adapted version of the “custom.region_if” example from
864 %1 = custom.region_if %0 : memref<?x?xf32> -> (memref<?x?xf32>)
887 %1 = custom.region_if %0 : memref<?x?xf32> -> (memref<?x?xf32>)
/external/llvm-project/mlir/test/lib/Dialect/Test/
DTestOps.td1833 def RegionIfOp : TEST_Op<"region_if",