Searched defs:gradient_uid (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 144 gradient_uid="__unsupported__"): argument
294 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
865 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
928 gradient_uid, argument
|
| D | control_flow_ops.py | 750 def EnterGradientColocation(self, op, gradient_uid): argument
755 def ExitGradientColocation(self, op, gradient_uid): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4349 def _colocate_with_for_gradient(self, op, gradient_uid, argument
5311 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): argument
|