| /external/tensorflow/tensorflow/python/debug/lib/ |
| H A D | debug_graph_reconstruction_test.py | 160 loop_cond = lambda i: math_ops.less(i, 16)
162 loop = control_flow_ops.while_loop(loop_cond, loop_body, [i])
|
| H A D | source_utils_test.py | 233 loop_cond = lambda i: math_ops.less(i, 16)
236 loop = control_flow_ops.while_loop(loop_cond, loop_body, [i])
|
| H A D | session_debug_testlib.py | 551 loop_cond = lambda i: math_ops.less(i, 16)
554 loop = control_flow_ops.while_loop(loop_cond, loop_body, [i])
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | functionalize_control_flow_test.cc | 208 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); local
210 ops::Switch(scope.WithOpName("while/Switch"), merge.output, loop_cond);
323 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); local
325 ops::Switch(scope.WithOpName("while/Switch"), merge.output, loop_cond);
438 auto loop_cond = ops::LoopCond(scope.WithOpName("while/LoopCond"), less); local
441 merge_x.output, loop_cond);
443 merge_y.output, loop_cond);
654 auto loop_cond = ops::LoopCond(scope.WithOpName("outer/LoopCond"), less_j); local
657 merge_j.output, loop_cond);
659 merge_k.output, loop_cond);
[all...] |
| H A D | functionalize_control_flow.cc | 75 Node* loop_cond = nullptr; member in struct:tensorflow::__anon25399::Frame
213 TF_ASSIGN_OR_RETURN(node_map[frame->loop_cond->id()],
219 return CopySubgraph(graph, frame, {frame->loop_cond}, squash_src_outputs,
334 if (frame->loop_cond == nullptr) {
341 for (const Edge* edge : frame->loop_cond->out_edges()) {
495 NodeDefBuilder builder(frame->loop_cond->name(), "XlaWhile");
1385 if (frame.loop_cond) {
1389 frame.loop_cond->name());
1391 frame.loop_cond = node;
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| H A D | profile_analyzer_cli_test.py | 333 loop_cond = lambda x: math_ops.less(x, 10)
339 loop = control_flow_ops.while_loop(loop_cond, loop_body, [x])
|
| /external/tensorflow/tensorflow/core/graph/ |
| H A D | graph_partition.cc | 478 const Edge* edge, Node* loop_cond,
507 Node* switch_node = AddControlSwitch(merge, loop_cond, device_name,
687 Node* loop_cond = cond_it->second; local
691 status = AddControlLoop(opts, g, src_frame, edge, loop_cond, &cf_info,
477 AddControlLoop(const PartitionOptions& opts, Graph* g, const Node* src, const Edge* edge, Node* loop_cond, std::vector<ControlFlowInfo>* cf_info, ControlLoop* loop) argument
|
| H A D | graph_partition_test.cc | 642 LoopCond loop_cond(scope, cv);
643 Switch switch_node(scope, merge.output, loop_cond);
656 enter, merge.output, cv, loop_cond, switch_node.output_false,
|
| /external/tensorflow/tensorflow/python/framework/ |
| H A D | meta_graph_test.py | 327 def loop_cond(it, _): function in function:ScopedMetaGraphTest._testScopedExport
335 loop_cond,
|
| /external/tensorflow/tensorflow/python/training/ |
| H A D | saver_test.py | 1954 def loop_cond(it, _): function in function:MetaGraphTest._testGraphExtensionSave
1962 loop_cond, loop_body,
2401 def loop_cond(it, _): function in function:ScopedGraphTest._testScopedSave
2408 _, biases2 = control_flow_ops.while_loop(loop_cond, loop_body, [
2473 def loop_cond(it, _): function in function:ScopedGraphTest._testScopedRestore
2480 _, biases = control_flow_ops.while_loop(loop_cond, loop_body, [
|
| /external/tensorflow/tensorflow/python/ops/ |
| H A D | control_flow_ops.py | 2653 self._pivot = loop_cond(pred, name="b_count")
2933 self._pivot = loop_cond(c, name="LoopCond")
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| H A D | control_flow_ops_py_test.py | 286 cond_op = control_flow_ops.loop_cond(less_op)
311 cond_op = control_flow_ops.loop_cond(less_op)
|