Searched defs:gradient_uid (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 143 gradient_uid="__unsupported__"): argument
292 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
872 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
935 gradient_uid, argument
|
| D | control_flow_ops.py | 760 def EnterGradientColocation(self, op, gradient_uid): argument
765 def ExitGradientColocation(self, op, gradient_uid): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4457 def _colocate_with_for_gradient(self, op, gradient_uid, argument
5419 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): argument
|