Searched defs:gradient_uid (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | gradients_util.py | 141 gradient_uid="__unsupported__"): argument
283 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
876 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
939 gradient_uid, argument
|
| H A D | control_flow_ops.py | 697 def EnterGradientColocation(self, op, gradient_uid): argument
702 def ExitGradientColocation(self, op, gradient_uid): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | ops.py | 4608 def _colocate_with_for_gradient(self, op, gradient_uid, argument
5578 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): argument
|