HomeSort by relevance Sort by last modified time
    Searched refs:switch_op (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
resolve_tensorflow_switch.cc 32 const auto* switch_op = switch_it->get(); local
33 if (switch_op->type != OperatorType::kSwitch) {
37 CHECK_EQ(switch_op->inputs.size(), 2);
38 CHECK_EQ(switch_op->outputs.size(), 2);
39 const string& predicate_name = switch_op->inputs[1];
45 LogName(*switch_op));
78 if (input == switch_op->outputs[selected_output_index]) {
79 input = switch_op->inputs[0];
95 if (*input_it == switch_op->outputs[nonselected_output_index]) {
110 if (!GetOpWithInput(*model, switch_op->outputs[i]))
    [all...]
  /external/tensorflow/tensorflow/python/framework/
auto_control_deps.py 171 def _process_switch(self, switch_op, ops_which_must_run,
200 switch_op: the switch op to be processed
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-access
226 for o in switch_op.outputs:
  /external/tensorflow/tensorflow/contrib/factorization/python/ops/
wals.py 42 switch_op):
59 switch_op: A TensorFlow op to be run before each sweep.
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)
wals_test.py 437 switch_op = control_flow_ops.group(
452 switch_op)
  /external/tensorflow/tensorflow/python/kernel_tests/
control_flow_ops_py_test.py 213 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...]

Completed in 796 milliseconds