Searched refs:_SwitchRefOrTensor (Results 1 – 3 of 3) 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_ops.py | 314 def _SwitchRefOrTensor(data, pred, name="Switch"): function
964 history_value = _SwitchRefOrTensor(history_value, pred)[branch]
1228 result = _SwitchRefOrTensor(result, pred)[1 - branch]
1236 val = _SwitchRefOrTensor(op.inputs[0], pred)[1 - branch]
1693 result = _SwitchRefOrTensor(result, self._pred)[self._branch]
1784 real_val = _SwitchRefOrTensor(real_val, self._pred)[self._branch]
2909 switch_vars = [_SwitchRefOrTensor(x, self._pivot) for x in merge_vars]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | control_flow_ops_py_test.py | 174 v1 = control_flow_ops._SwitchRefOrTensor(v._ref(), p) # pylint: disable=protected-access
|