Home
last modified time | relevance | path

Searched refs:switch_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dresolve_tensorflow_switch.cc32 const auto* switch_op = switch_it->get(); in Run() local
33 if (switch_op->type != OperatorType::kSwitch) { in Run()
37 CHECK_EQ(switch_op->inputs.size(), 2); in Run()
38 CHECK_EQ(switch_op->outputs.size(), 2); in Run()
39 const string& predicate_name = switch_op->inputs[1]; in Run()
45 LogName(*switch_op)); in Run()
78 if (input == switch_op->outputs[selected_output_index]) { in Run()
79 input = switch_op->inputs[0]; in Run()
95 if (*input_it == switch_op->outputs[nonselected_output_index]) { in Run()
110 if (!GetOpWithInput(*model, switch_op->outputs[i])) { in Run()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dauto_control_deps.py171 def _process_switch(self, switch_op, ops_which_must_run, argument
207 inp = switch_op.inputs[0]
211 if switch_op.outputs[0] in merge_for_resource:
213 new_merge = control_flow_ops.merge(switch_op.outputs,
216 switch_op._control_flow_context.outer_context) # pylint: disable=protected-access
221 …switch_op._add_control_input(last_op_using_resource_tensor[inp]) # pylint: disable=protected-acce…
226 for o in switch_op.outputs:
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals.py42 switch_op): argument
68 self._switch_op = switch_op
290 switch_op = control_flow_ops.group(
303 row_prep_ops, col_prep_ops, row_train_op, col_train_op, switch_op)
Dwals_test.py437 switch_op = control_flow_ops.group(
452 switch_op)
/external/tensorflow/tensorflow/python/kernel_tests/
Dcontrol_flow_ops_py_test.py213 switch_op = control_flow_ops.switch(data, pred)
214 merge_op = control_flow_ops.merge(switch_op)[0]
226 switch_op = control_flow_ops.switch(data, ports)
227 dead_branch = array_ops.identity(switch_op[0])
241 switch_op = control_flow_ops.switch(data, less_op)
242 merge_op = control_flow_ops.merge(switch_op)[0]
252 switch_op = control_flow_ops.switch(data, ports)
254 add_op = math_ops.add(switch_op[0], one)
255 id_op = array_ops.identity(switch_op[1])
266 switch_op = control_flow_ops.switch(data, ports)
[all …]