Home
last modified time | relevance | path

Searched refs:is_stateless (Results 1 – 25 of 68) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dmerge_control_flow.mlir18 }) {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 …]
Dfunctional-control-flow-to-regions.mlir12 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 …]
Dregion-control-flow-to-functional.mlir20 …}) {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 …]
Ddrop_while_shape_invariant.mlir6 …%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>)
Dmark_ops_for_outside_compilation.mlir191 }) {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 …]
Dfunctional-control-flow-to-cfg.mlir10 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
Dresource_op_lifting.mlir162 {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 …]
Dresource-device-inference.mlir62 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}
Dtpu_extract_outside_compilation.mlir343 }) { _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 …]
Dwhile_licm.mlir24 …) {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>)
Dresource-alias-analysis-test.mlir86 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/
Dfunctional-while-ops.mlir5 …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
Dfunctional-if-ops.mlir6 …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
Dcase.mlir8 … = [@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/
Dwhile_op.mlir10 … @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/
Dwhile_op.mlir18 … @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/
Dwhile_op.mlir18 … @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/
Dfunctional-while-ops.pbtxt4 # 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…
Dfunctional-if-ops.pbtxt4 # 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/
Dif_op.mlir4 // 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/
Doptimize_functional_ops.mlir12 …%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/
Dtf_ops.cc175 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/
Dlegalize-tf-types.mlir16 // CHECK-NEXT: }) {is_stateless = false} : (tensor<i1>) -> tensor<1xi8>
22 }) {is_stateless = false} : (tensor<i1>) -> tensor<1x!tf.qint8>
Dlegalize-tf-control-flow.mlir24 …%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/
Dfunctional_control_flow_to_regions.cc100 if_op.getLoc(), if_op.getResultTypes(), cond, if_op.is_stateless(), in ConvertIfOp()
119 while_op.parallel_iterations(), while_op.is_stateless(), in ConvertWhileOp()

123