Home
last modified time | relevance | path

Searched refs:NextIteration (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dtf_executor_ops_side_effect.mlir3 // NextIteration.Source and NextIteration.Sink.
5 // CHECK: tf_executor.NextIteration.Source
6 // CHECK: tf_executor.NextIteration.Source
7 // CHECK: tf_executor.NextIteration.Sink
8 // CHECK: tf_executor.NextIteration.Sink
11 …%output_1, %token_1, %control_1 = tf_executor.NextIteration.Source : tensor<1xi32> {T = i32, devic…
12 …%output_2, %token_2, %control_2 = tf_executor.NextIteration.Source : tensor<1xi32> {T = i32, devic…
13 tf_executor.NextIteration.Sink [%token_1] %output_1 : tensor<1xi32> {T = i32, device = ""}
14 tf_executor.NextIteration.Sink [%token_2] %output_2 : tensor<1xi32> {T = i32, device = ""}
Dgraph_pruning.mlir42 // Check that NextIteration.sink node isn't deleted when the source is still
46 // CHECK: tf_executor.NextIteration.Source
47 // CHECK: tf_executor.NextIteration.Sink
49 %1:3 = tf_executor.NextIteration.Source : i32
50 tf_executor.NextIteration.Sink[%1#1] %1#0 : i32
56 // Check that NextIteration.sink node is deleted when the source does not have
60 // CHECK-NOT: tf_executor.NextIteration.Source
61 // CHECK-NOT: tf_executor.NextIteration.Sink
63 %1:3 = tf_executor.NextIteration.Source : i32
65 tf_executor.NextIteration.Sink[%1#1] %1#0 : i32
[all …]
Dtpu_device_propagation.mlir261 // Tests tf_executor.NextIteration.Source/tf_executor.NextIteration.Sink has a
267 // CHECK: tf_executor.NextIteration.Source
269 %0:3 = tf_executor.NextIteration.Source : tensor<i64> {T = "tfdtype$DT_INT64"}
276 // CHECK: tf_executor.NextIteration.Sink
278 tf_executor.NextIteration.Sink [%0#1] %2#0 : tensor<i64> {T = "tfdtype$DT_INT64"}
284 // Tests tf_executor.NextIteration with mismatched devices does not propagate
290 // CHECK: tf_executor.NextIteration.Source
292 …%0:3 = tf_executor.NextIteration.Source : tensor<i64> {device = "/job:localhost/replica:0/task:0/d…
295 // CHECK: tf_executor.NextIteration.Sink
297 …tf_executor.NextIteration.Sink [%0#1] %1#0 : tensor<i64> {device = "/job:localhost/replica:0/task:…
[all …]
Dtf_executor_ops_invalid.mlir661 // Check that a tf_executor.NextIteration.Sink parent is a graph.
664 tf_executor.NextIteration.Sink[%arg1] %arg0 : tensor<*xf32>
665 // expected-error@-1 {{'tf_executor.NextIteration.Sink' op expects parent op 'tf_executor.graph'}}
672 // Check that a tf_executor.NextIteration.Source parent is a graph.
675 %1:3 = tf_executor.NextIteration.Source : tensor<*xf32>
676 // expected-error@-1 {{'tf_executor.NextIteration.Source' op expects parent op 'tf_executor.graph'}}
685 %1:3 = tf_executor.NextIteration.Source : tensor<*xf32>
686 // expected-error@-1 {{'tf_executor.NextIteration.Source' op expects a single user for produced tok…
696 %1:3 = tf_executor.NextIteration.Source : tensor<*xf32>
697 // expected-error@-1 {{'tf_executor.NextIteration.Source' op token should be consumed by a sink op}}
[all …]
Dtf_executor_ops.mlir433 %1:3 = tf_executor.NextIteration.Source : tensor<*xf32>
434 tf_executor.NextIteration.Sink[%1#1] %1#0 : tensor<*xf32>
435 // CHECK: tf_executor.NextIteration.Source : tensor<*xf32>
436 // CHECK: tf_executor.NextIteration.Sink[%{{.*}}] %{{.*}} : tensor<*xf32>
445 …%1:3 = tf_executor.NextIteration.Source : tensor<*xf32> {attr3 = 32 : i64, tf_executor.attr_fetch …
446 …tf_executor.NextIteration.Sink[%1#1] %1#0 : tensor<*xf32> {attr4 = 42 : i64, tf_executor.attr_push…
447 // CHECK: tf_executor.NextIteration.Source : tensor<*xf32> {attr3 = 32 : i64, tf_executor.attr_fetc…
448 // CHECK: tf_executor.NextIteration.Sink[%{{.*}}] %{{.*}} : tensor<*xf32> {attr4 = 42 : i64, tf_exe…
459 %3:3 = tf_executor.NextIteration.Source : tensor<*xf32>
460 tf_executor.NextIteration.Sink[%3#1] %3#0, %1#2 : tensor<*xf32>
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dmultiple-use-next-iteration.pbtxt3 # Verify that a NextIteration node feeding two different merge nodes is properly
7 # CHECK: %[[NEXTITERATION:[a-z_0-9]+]], {{.*}} = tf_executor.NextIteration.Source
65 input: "NextIteration"
79 # The following node also takes NextIteration node as input.
84 input: "NextIteration"
132 name: "NextIteration"
133 op: "NextIteration"
Dgraph-while-loop.pbtxt3 # Verify that importing a Graph with a backedge leads to two NextIteration nodes
7 …TERATION:[a-z0-9]+]], %[[NEXTITERATION_token:[a-z0-9]+]], {{.*}} = tf_executor.NextIteration.Source
10 # CHECK: tf_executor.NextIteration.Sink[%[[NEXTITERATION_token]]]
66 input: "while/NextIteration"
185 name: "while/NextIteration"
186 op: "NextIteration"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/mlir2graphdef/
Dwhile-loop.mlir7 // CHECK-NEXT: input: "while/NextIteration"
8 // CHECK: name: "while/NextIteration"
9 // CHECK-NEXT: op: "NextIteration"
12 …%0:3 = tf_executor.NextIteration.Source : tensor<*xi32> {device = "", T = "tfdtype$DT_INT32"} loc(…
23 …tf_executor.NextIteration.Sink [%0#1] %11#0 : tensor<*xi32> {device = "", T = "tfdtype$DT_INT32"} …
Dref-while-loop.mlir11 …%0:3 = tf_executor.NextIteration.Source : tensor<*x!tf.int32ref> {device = "", T = "tfdtype$DT_INT…
22 …tf_executor.NextIteration.Sink [%0#1] %11#0 : tensor<*x!tf.int32ref> {device = "", T = "tfdtype$DT…
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_executor_ops.td41 last returned value (exceptions are `tf_executor.NextIteration.graph`,
42 `tf_executor.NextIteration.sink` and `tf_executor.fetch` which don’t return any
95 (`tf_executor.NextIteration.Source` is the exception).
417 def TfExecutor_NextIterationSourceOp : TfExecutor_Op<"NextIteration.Source",
420 The "tf_executor.NextIteration.Source" is paired with a
421 "tf_executor.NextIteration.sink" to represent NextIteration op in
426 Tensorflow NextIteration operation forwards its input to the next iteration
427 of a while loop. Each loop variable needs its own NextIteration op.
432 In the TF executor dialect, the NextIteration op is broken into
433 tf_executor.NextIteration.sink and tf_executor.NextIteration.source because
[all …]
/external/tensorflow/tensorflow/core/grappler/costs/graph_properties_testdata/
Dnested_loop.pbtxt114 input: "while/NextIteration"
331 input: "while/while/NextIteration"
546 name: "while/while/NextIteration"
547 op: "NextIteration"
558 op: "NextIteration"
671 name: "while/NextIteration"
672 op: "NextIteration"
683 op: "NextIteration"
Dloops_and_resource_vars.pbtxt189 input: "while/NextIteration"
459 input: "while/while/NextIteration"
639 name: "while/while/NextIteration"
640 op: "NextIteration"
651 op: "NextIteration"
717 name: "while/NextIteration"
718 op: "NextIteration"
729 op: "NextIteration"
Dloops_and_queues.pbtxt151 input: "while/NextIteration"
484 input: "while/while/NextIteration"
699 name: "while/while/NextIteration"
700 op: "NextIteration"
711 op: "NextIteration"
777 name: "while/NextIteration"
778 op: "NextIteration"
789 op: "NextIteration"
Dwhile_loop.pbtxt107 input: "while/NextIteration"
322 name: "while/NextIteration"
323 op: "NextIteration"
334 op: "NextIteration"
Dwhile_loop_var_handle_op.pbtxt107 input: "while/NextIteration"
256 name: "while/NextIteration"
257 op: "NextIteration"
266 op: "NextIteration"
Dqueues_and_loops.pbtxt224 input: "while/NextIteration"
439 name: "while/NextIteration"
440 op: "NextIteration"
451 op: "NextIteration"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/g3doc/
Dtf_dialects.md90 NextIteration operation.
119 (`tf_executor.NextIteration.Source` is the exception).
126 last returned value (exceptions are `tf_executor.NextIteration.sink` and
242 ### NextIteration: `tf_executor.NextIteration.Source` and `tf_executor.NextIteration.Sink` Operation
245 [`NextIteration`](https://www.tensorflow.org/api_docs/cc/class/tensorflow/ops/next-iteration)
249 DAG.`tf_executor.NextIteration.Source`does not take any operand and produces two
252 paired`tf_executor.NextIteration.Sink`Operation alongside the value that is
254 by`tf_executor.NextIteration.Sink`. The type of the result of the source must
257 `tf_executor.NextIteration.Source` is an exception in the executor model in the
258 sense that it executes after the paired `tf_executor.NextIteration.Sink` even
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dscc_test.cc242 input: "while/NextIteration" in TEST_F()
361 name: "while/NextIteration" in TEST_F()
362 op: "NextIteration" in TEST_F()
/external/tensorflow/tensorflow/core/api_def/java_api/
Dapi_def_NextIteration.pbtxt2 graph_op_name: "NextIteration"
/external/tensorflow/tensorflow/core/api_def/python_api/
Dapi_def_NextIteration.pbtxt2 graph_op_name: "NextIteration"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DNextIteration.pbtxt2 name: "NextIteration"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DNextIteration.pbtxt2 name: "NextIteration"
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_NextIteration.pbtxt2 graph_op_name: "NextIteration"
/external/tensorflow/tensorflow/compiler/jit/
Dshape_inference_test.cc104 ops::NextIteration(scope.WithOpName("while/NextIteration"), add); in TEST()
150 ops::NextIteration(scope.WithOpName("while/NextIteration"), identity); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_constructor_test.cc339 input: "while/NextIteration" in TEST_F()
458 name: "while/NextIteration" in TEST_F()
459 op: "NextIteration" in TEST_F()
609 input: "while/NextIteration" in TEST_F()
761 name: "while/NextIteration" in TEST_F()
762 op: "NextIteration" in TEST_F()
773 op: "NextIteration" in TEST_F()
2110 input: "while/NextIteration" in TEST_F()
2229 name: "while/NextIteration" in TEST_F()
2230 op: "NextIteration" in TEST_F()

123