Home
last modified time | relevance | path

Searched refs:switch_i (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/cc/ops/
Dwhile_loop.cc203 auto switch_i = Switch(scope, merge_outputs[i], cond_out); in BuildWhileLoop() local
204 switch_trues[i] = switch_i.output_true; in BuildWhileLoop()
205 switch_falses[i] = switch_i.output_false; in BuildWhileLoop()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc963 auto switch_i = ops::Switch(scope.WithOpName("outer/Switch"), in RunTest() local
966 switch_i.output_false); in RunTest()
968 ops::Identity(scope.WithOpName("outer/Identity"), switch_i.output_true); in RunTest()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py336 switch_i = control_flow_ops.switch(merge_i, cond_op)
338 add_i = math_ops.add(switch_i[1], enter_one)
343 exit_i = control_flow_ops.exit(switch_i[0])
362 switch_i = control_flow_ops.switch(merge_i, cond_op)
364 add_i = math_ops.add(switch_i[1], enter_one)
370 exit_i = control_flow_ops.exit(switch_i[0])