/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
D | merge_control_flow.mlir | 18 }) {is_stateless = true} : (tensor<i1>) -> () 24 }) {is_stateless = true} : (tensor<i1>) -> () 49 }) {is_stateless = true} : (tensor<i1>) -> () 58 }) {is_stateless = true} : (tensor<i1>) -> () 60 }) {is_stateless = false} : (tensor<i32>, tensor<?xf32>) -> (tensor<i32>, tensor<?xf32>) 83 …}) {is_stateless = true, _else_func_name = "elseFunc1", _then_func_name = "thenFunc1"} : (tensor<i… 89 …}) {is_stateless = true, _else_func_name = "elseFunc2", _then_func_name = "thenFunc2"} : (tensor<i… 110 }) {is_stateless = true} : (tensor<i1>) -> (tensor<f32>) 118 }) {is_stateless = true} : (tensor<i1>) -> () 139 }) {is_stateless = true} : (tensor<i1>) -> (tensor<f32>) [all …]
|
D | functional-control-flow-to-regions.mlir | 12 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false, 27 // CHECK-SAME: is_stateless = false 45 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 69 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 91 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 115 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 139 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 159 is_stateless = true, 173 // CHECK-SAME: is_stateless = true 189 is_stateless = false [all …]
|
D | region-control-flow-to-functional.mlir | 20 …}) {is_stateless = true, _attr0 = false, attr1 = "hello", _then_func_name = "test_then_name", _els… 43 …}) {is_stateless = true, _attr0 = false, attr1 = "hello", _then_func_name = "", _else_func_name = … 66 }) { is_stateless = true} : (tensor<i1>) -> tensor<2xf32> 86 }) { is_stateless = true} : (tensor<i1>) -> tensor<2xf32> 121 }) { is_stateless = true } : (tensor<i1>) -> tensor<*xf32> 129 }) { is_stateless = true } : (tensor<i1>) -> tensor<*xf32> 146 }) {is_stateless = false} : (tensor<i1>) -> tensor<*xf32> 166 }) {is_stateless = false} : (tensor<i1>) -> tensor<2xf32> 188 }) {is_stateless = false} : (tensor<i1>) -> tensor<2xf32> 208 }) {is_stateless = false} : (tensor<i1>) -> tensor<2xf32> [all …]
|
D | drop_while_shape_invariant.mlir | 6 …%0 = "tf.While"(%arg0) {cond = @while_cond, body = @while_body, is_stateless = false, shape_invari… 16 }) {is_stateless = false, shape_invariant} : (tensor<4xf32>) -> (tensor<*xf32>)
|
D | mark_ops_for_outside_compilation.mlir | 191 }) {is_stateless = true} : (tensor<i1>) -> (tensor<f32>) 219 // CHECK: {is_stateless 220 }) {is_stateless = true} : (tensor<i1>) -> (tensor<f32>) 257 // CHECK: {is_stateless 258 }){is_stateless = true} : (tensor<i1>) -> (tensor<f32>) 260 // CHECK: {is_stateless 261 }) {is_stateless = true} : (tensor<i1>) -> (tensor<f32>) 291 }) {is_stateless = true} : (tensor<f32>, tensor<i32>) -> (tensor<f32>, tensor<i32>) 324 // CHECK: {is_stateless = true 325 }) {is_stateless = true} : (tensor<f32>, tensor<i32>) -> (tensor<f32>, tensor<i32>) [all …]
|
D | functional-control-flow-to-cfg.mlir | 10 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 35 then_branch = @testIf3Then, else_branch = @testIf3Else, is_stateless = false 64 then_branch = @testIfThen, else_branch = @testIfElse, is_stateless = false 98 then_branch = @testIf1Then, else_branch = @testIf1Else, is_stateless = false 114 cond = @testWhile2Cond, body = @testWhile2Body, is_stateless = false 140 "tf.While"() { cond = @testWhile0Cond, body = @testWhile0Body, is_stateless = false } : () -> () 163 cond = @testWhile2Cond, body = @testWhile2Body, is_stateless = false 201 cond = @testWhileCond, body = @testWhileBody, is_stateless = false
|
D | resource_op_lifting.mlir | 162 {body = @while_body, cond = @while_cond, device = "", is_stateless = false} 211 body = @while_body, cond = @while_cond, device = "", is_stateless = false} 252 body = @while_body, cond = @while_cond, device = "", is_stateless = false} 290 body = @while_body, cond = @while_cond, device = "", is_stateless = false} 306 body = @while_body1, cond = @while_cond1, device = "", is_stateless = false} 345 body = @while_body, cond = @while_cond, device = "", is_stateless = false} 369 body = @while_body, cond = @while_cond, device = "", is_stateless = false} 400 …%3:2 = "tf.Case"(%arg0, %0, %1) {branches = [@branch_0, @branch_1, @branch_2], is_stateless = fals… 458 is_stateless = false} 506 is_stateless = false} [all …]
|
D | resource-device-inference.mlir | 62 is_stateless = false} 134 cond = @while_cond, is_stateless = false} 215 else_branch = @if_then_and_else, is_stateless = false} 222 is_stateless = false} 320 }) {is_stateless = false} : (tensor<i1>) -> () 420 }) {is_stateless = false} : (tensor<i1>) -> () 464 }){is_stateless = false} 495 }){is_stateless = false}
|
D | tpu_extract_outside_compilation.mlir | 343 }) { _xla_outside_compilation = "cluster1", is_stateless = false} : (tensor<i1>) -> () 600 …}) { is_stateless = false, _then_func_name = "test_then_name", _else_func_name = "test_else_name"}… 648 }) { is_stateless = false} : (tensor<i1>) -> () 655 }) { is_stateless = false} : (tensor<i1>) -> () 705 }) {_xla_outside_compilation = "cluster1", is_stateless = false} : (tensor<i1>) -> () 775 }) {_xla_outside_compilation = "cluster1", is_stateless = false} : (tensor<i1>) -> () 780 }) {is_stateless = false} : (tensor<i1>) -> () 838 }) { is_stateless = false} : (tensor<i1>) -> (tensor<?xi32>) 884 }) { is_stateless = false} : (tensor<i1>) -> () 950 }) { is_stateless = false} : (tensor<i1>) -> () [all …]
|
D | while_licm.mlir | 24 …) {is_stateless = false} : (tensor<i32>, tensor<1xf32>) -> (tensor<i32>, tensor<1xf32>) loc("While… 66 ) { is_stateless = false } : (tensor<4xf32>, tensor<i32>) -> (tensor<4xf32>, tensor<i32>)
|
D | resource-alias-analysis-test.mlir | 86 then_branch = @if_then, else_branch = @if_else, is_stateless = true 126 is_stateless = true 170 body = @while_body, cond = @while_cond, is_stateless = false 244 }) {is_stateless = true} : (tensor<i1>) -> (!tf_res, !tf_res, !tf_res) 275 }) {is_stateless = true} : (tensor<i32>) -> (!tf_res, !tf_res, !tf_res) 308 }) {is_stateless = false} : (!tf_res, !tf_res, !tf_res) -> (!tf_res, !tf_res, !tf_res)
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/ |
D | functional-while-ops.mlir | 5 …or.island wraps "tf.While"(%arg0, %arg1) {body = @body, cond = @cond, is_stateless = false} : (ten… 6 …or.island wraps "tf.While"(%arg0, %arg1) {body = @body, cond = @cond, is_stateless = true} : (tens… 7 …or.island wraps "tf.While"(%arg0, %arg1) {body = @body, cond = @cond, is_stateless = false, shape_… 33 // is_stateless attribute is present and otherwise it is mapped to TensorFlow 39 // CHECK-NOT: is_stateless 58 // CHECK-NOT: is_stateless 76 // CHECK-NOT: is_stateless
|
D | functional-if-ops.mlir | 6 …d wraps "tf.If"(%outputs_2, %arg2, %arg3) {else_branch = @cond_false, is_stateless = false, then_b… 7 …d wraps "tf.If"(%outputs_2, %arg2, %arg3) {else_branch = @cond_false, is_stateless = true, then_br… 29 // Verify that If op is mapped to TensorFlow StatelessIf op if the is_stateless 36 // CHECK-NOT: is_stateless 53 // CHECK-NOT: is_stateless
|
D | case.mlir | 8 … = [@indexed_case_branch0_40, @indexed_case_branch1_50], device = "", is_stateless = true, output_… 10 … = [@indexed_case_branch0_40, @indexed_case_branch1_50], device = "", is_stateless = false, output…
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/ |
D | while_op.mlir | 10 … @while_body_with_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, parall… 11 …_with_wrong_cluster_attr, cond = @while_cond_with_wrong_cluster_attr, is_stateless = false, parall… 12 …hile_body_without_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, parall… 13 …e_body_without_cluster_attr, cond = @while_cond_without_cluster_attr, is_stateless = false, parall… 14 …hile_body_with_cluster_attr, cond = @while_cond_without_cluster_attr, is_stateless = false, parall…
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/ |
D | while_op.mlir | 18 … @while_body_with_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, name =… 19 …hile_body_without_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, name =… 20 …hile_body_with_cluster_attr, cond = @while_cond_without_cluster_attr, is_stateless = false, name =…
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_island_inlining/ |
D | while_op.mlir | 18 … @while_body_with_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, name =… 19 …hile_body_without_cluster_attr, cond = @while_cond_with_cluster_attr, is_stateless = false, name =… 20 …hile_body_with_cluster_attr, cond = @while_cond_without_cluster_attr, is_stateless = false, name =…
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/ |
D | functional-while-ops.pbtxt | 4 # composite While op in MLIR with is_stateless attribute set accordingly to 7 # CHECK-DAG: "tf.While"{{.*}} is_stateless = false{{.*}} loc("StatefulWhile") 8 # CHECK-DAG: "tf.While"{{.*}} is_stateless = true{{.*}} loc("StatelessWhile") 9 # CHECK-DAG: "tf.While"{{.*}} is_stateless = false{{.*}} shape_invariant{{.*}} -> (tensor<i32>, ten…
|
D | functional-if-ops.pbtxt | 4 # composite If op in MLIR with is_stateless attribute set accordingly to 7 # CHECK-DAG: "tf.If"{{.*}} is_stateless = false{{.*}} loc("StatefulIf") 8 # CHECK-DAG: "tf.If"{{.*}} is_stateless = true{{.*}} loc("StatelessIf")
|
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
D | if_op.mlir | 4 // CHECK: %{{.*}} = "tf.If"(%{{.*}}, %{{.*}}, %{{.*}}) {else_branch = @cond_false, is_stateless =… 6 …%1 = "tf.If"(%0, %arg0, %arg1) {else_branch = @cond_false, then_branch = @cond_true, is_stateless …
|
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
D | optimize_functional_ops.mlir | 12 …%3 = "tf.If"(%2, %0, %1) {else_branch = @sub, then_branch = @add, is_stateless = true} : (tensor<i… 39 …%3 = "tf.If"(%2, %0, %1) {else_branch = @sub, then_branch = @addormul, is_stateless = true} : (ten… 45 …%1 = "tf.If"(%0, %arg1, %arg0) {else_branch = @mul, then_branch = @add, is_stateless = true} : (te… 78 else_branch = @_functionalize_if_else_branch_00, is_stateless = false, 111 else_branch = @_functionalize_if_else_branch_01, is_stateless = false, 147 else_branch = @_functionalize_if_else_branch_02, is_stateless = true,
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_ops.cc | 175 if (auto is_stateless = op->getAttrOfType<BoolAttr>("is_stateless")) in CanDuplicate() local 176 return is_stateless.getValue(); in CanDuplicate() 190 if (auto is_stateless = op->getAttrOfType<BoolAttr>("is_stateless")) in CanHaveSideEffects() local 191 return !is_stateless.getValue(); in CanHaveSideEffects()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
D | legalize-tf-types.mlir | 16 // CHECK-NEXT: }) {is_stateless = false} : (tensor<i1>) -> tensor<1xi8> 22 }) {is_stateless = false} : (tensor<i1>) -> tensor<1x!tf.qint8>
|
D | legalize-tf-control-flow.mlir | 24 …%1 = "tf.If"(%0, %arg0, %arg1) {else_branch = @cond_false, is_stateless = true, then_branch = @con… 69 }) {is_stateless = true} : (tensor<i1>) -> tensor<f32> 79 …%0:2 = "tf.Case"(%index, %arg0, %arg1) {branches = [@exponential, @log, @floor], is_stateless = tr… 151 }) {is_stateless = true} : (tensor<i32>) -> (tensor<f32>, tensor<f32>) 184 …%2:3 = "tf.While"(%0, %1, %0) {body = @while_body, cond = @while_cond, is_stateless = true, parall… 235 …}) {is_stateless = true, parallel_iterations = 10 : i64} : (tensor<i32>, tensor<i32>, tensor<i32>)… 277 }) {is_stateless = true, parallel_iterations = 10 : i64} : (tensor<i32>) -> tensor<i32> 310 }) {is_stateless = true, parallel_iterations = 10 : i64} : () -> ()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | functional_control_flow_to_regions.cc | 100 if_op.getLoc(), if_op.getResultTypes(), cond, if_op.is_stateless(), in ConvertIfOp() 119 while_op.parallel_iterations(), while_op.is_stateless(), in ConvertWhileOp()
|