| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 103 auto next_iteration = in TEST() local
110 scope.graph()->AddEdge(next_iteration.node(), 0, merge.output.node(), 1); in TEST()
149 auto next_iteration = in TEST() local
155 scope.graph()->AddEdge(next_iteration.node(), 0, merge.output.node(), 1); in TEST()
|
| D | xla_cluster_util_test.cc | 262 Output next_iteration = in MakeLoop() local
264 return s.graph()->UpdateEdge(next_iteration.node(), 0, merge.output.node(), in MakeLoop()
|
| D | resource_operation_safety_analysis_test.cc | 496 Output next_iteration = in TEST() local
499 root.graph()->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1)); in TEST()
506 root.graph()->AddControlEdge(read, next_iteration.node()); in TEST()
|
| D | deadness_analysis_test.cc | 148 Output next_iteration = in CreateInductionVariable() local
152 ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1) in CreateInductionVariable()
211 Output next_iteration = ops::NextIteration( in CreateDependentLoopInvariantValue() local
214 ->UpdateEdge(next_iteration.node(), 0, iv.output.node(), 1) in CreateDependentLoopInvariantValue()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_while.cc | 200 node_map[arg.next_iteration->id()] = retval_node; in BuildLoopBody()
201 next_iterations.push_back(arg.next_iteration); in BuildLoopBody()
335 &arg.next_iteration)); in FunctionalizeLoop()
336 if (!IsNextIteration(arg.next_iteration)) { in FunctionalizeLoop()
339 FormatNodeForError(*arg.next_iteration), " with kind ", in FunctionalizeLoop()
340 arg.next_iteration->type_string()); in FunctionalizeLoop()
|
| D | functionalize_control_flow_util.h | 44 Node* next_iteration = nullptr; member
|
| D | functionalize_control_flow_test.cc | 339 auto next_iteration = in TEST() local
346 scope.graph()->AddEdge(next_iteration.node(), 0, merge.output.node(), 1); in TEST()
617 auto next_iteration = in TEST() local
622 scope.graph()->AddEdge(next_iteration.node(), 0, merge.output.node(), 1); in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | lower_while_op.cc | 443 Node* next_iteration; in CreateNextIterationNodes() local
454 .Finalize(graph_, &next_iteration)); in CreateNextIterationNodes()
455 next_iterations_nodes_.emplace_back(next_iteration); in CreateNextIterationNodes()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | xla_hlo_profile_test.cc | 293 auto next_iteration = in XLA_TEST_F() local
295 Tuple(&builder, {next_iteration, Mul(matrix, matrix)}); in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 652 NextIteration next_iteration(scope, identity); in TEST() local
659 scope.graph()->AddEdge(next_iteration.node(), 0, merge.output.node(), 1); in TEST()
664 identity, next_iteration, while_exits.back()}) { in TEST()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 314 next_n = control_flow_ops.next_iteration(switch_n[0])
340 next_i = control_flow_ops.next_iteration(add_i)
367 next_i = control_flow_ops.next_iteration(add_i)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_ops.py | 215 return next_iteration(data, name=name)
|