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.cc809 auto switch_i = ops::Switch(scope.WithOpName("outer/Switch"), in TEST() local
812 switch_i.output_false); in TEST()
814 ops::Identity(scope.WithOpName("outer/Identity"), switch_i.output_true); in TEST()
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py310 switch_i = control_flow_ops.switch(merge_i, cond_op)
312 add_i = math_ops.add(switch_i[1], enter_one)
317 exit_i = control_flow_ops.exit(switch_i[0])
336 switch_i = control_flow_ops.switch(merge_i, cond_op)
338 add_i = math_ops.add(switch_i[1], enter_one)
344 exit_i = control_flow_ops.exit(switch_i[0])