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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
resolve_tensorflow_switch.cc 29 const auto* switch_op = switch_it->get(); local
30 if (switch_op->type != OperatorType::kTensorFlowSwitch) {
34 CHECK_EQ(switch_op->inputs.size(), 2);
35 CHECK_EQ(switch_op->outputs.size(), 2);
36 const string& predicate_name = switch_op->inputs[1];
42 LogName(*switch_op));
75 if (input == switch_op->outputs[selected_output_index]) {
76 input = switch_op->inputs[0];
92 if (*input_it == switch_op->outputs[nonselected_output_index]) {
105 if (!GetOpWithInput(*model, switch_op->outputs[i]))
    [all...]
  /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 435 switch_op = control_flow_ops.group(
450 switch_op)
  /external/tensorflow/tensorflow/python/eager/
function.py 858 def _process_switch(self, switch_op, ops_which_must_run,
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
control_flow_ops_py_test.py 196 switch_op = control_flow_ops.switch(data, pred)
197 merge_op = control_flow_ops.merge(switch_op)[0]
208 switch_op = control_flow_ops.switch(data, ports)
209 dead_branch = array_ops.identity(switch_op[0])
222 switch_op = control_flow_ops.switch(data, less_op)
223 merge_op = control_flow_ops.merge(switch_op)[0]
232 switch_op = control_flow_ops.switch(data, ports)
234 add_op = math_ops.add(switch_op[0], one)
235 id_op = array_ops.identity(switch_op[1])
245 switch_op = control_flow_ops.switch(data, ports
    [all...]

Completed in 318 milliseconds