Searched refs:_SwitchRefOrTensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | control_flow_grad.py | 106 return control_flow_ops._SwitchRefOrTensor(grad, grad_ctxt.pivot)
130 return control_flow_ops._SwitchRefOrTensor(grad, pred, name="cond_grad")
136 return [control_flow_ops._SwitchRefOrTensor(grad, cond[i])[1]
|
| D | control_flow_state.py | 425 history_value = control_flow_ops._SwitchRefOrTensor(
687 result = control_flow_ops._SwitchRefOrTensor(result, pred)[1 - branch]
695 val = control_flow_ops._SwitchRefOrTensor(op.inputs[0],
|
| D | control_flow_ops.py | 329 def _SwitchRefOrTensor(data, pred, name="Switch"): function
961 result = _SwitchRefOrTensor(result, self._pred)[self._branch]
1052 real_val = _SwitchRefOrTensor(real_val, self._pred)[self._branch]
2209 switch_vars = [_SwitchRefOrTensor(x, self._pivot) for x in merge_vars]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 200 v1 = control_flow_ops._SwitchRefOrTensor(v._ref(), p) # pylint: disable=protected-access
|