Searched defs:gradient_uid (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 194 gradient_uid="__unsupported__"): argument
356 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
911 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
961 gradient_uid, argument
|
| D | control_flow_ops.py | 1475 def EnterGradientColocation(self, op, gradient_uid): argument
1480 def ExitGradientColocation(self, op, gradient_uid): argument
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 212 def EnterGradientColocation(self, op, gradient_uid): argument
251 def ExitGradientColocation(self, op, gradient_uid): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4259 def _colocate_with_for_gradient(self, op, gradient_uid, argument
5196 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): argument
|