Searched refs:switch_op (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
D | resolve_tensorflow_switch.cc | 32 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/ |
D | auto_control_deps.py | 171 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/ |
D | wals.py | 42 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)
|
D | wals_test.py | 437 switch_op = control_flow_ops.group( 452 switch_op)
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | 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 …]
|