Searched refs:gradient_uid (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 144 gradient_uid="__unsupported__"): argument
168 with _maybe_colocate_with(y.op, gradient_uid, colocate_gradients_with_ops):
294 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
297 with ops._colocate_with_for_gradient(op, gradient_uid): # pylint: disable=protected-access
527 gradient_uid = ops.get_default_graph().unique_name("uid")
537 gradient_uid)
590 with _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops):
593 out_grads = _AggregatedGrads(grads, op, gradient_uid, loop_state,
697 gradient_uid,
865 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
[all …]
|
| D | control_flow_ops.py | 750 def EnterGradientColocation(self, op, gradient_uid): argument
753 self._outer_context.EnterGradientColocation(op, gradient_uid)
755 def ExitGradientColocation(self, op, gradient_uid): argument
758 self._outer_context.ExitGradientColocation(op, gradient_uid)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 393 def EnterGradientColocation(self, op: ops.Operation, gradient_uid: Text):
405 cluster = parts[0] + "." + gradient_uid
418 if gradient_uid == "__unsupported__":
441 cluster = parts[0] + "." + gradient_uid
447 def ExitGradientColocation(self, op: ops.Operation, gradient_uid: Text):
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4349 def _colocate_with_for_gradient(self, op, gradient_uid, argument
4352 if gradient_uid is not None:
4355 ctx.EnterGradientColocation(op, gradient_uid)
4359 ctx.ExitGradientColocation(op, gradient_uid)
5311 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): argument
5328 op, gradient_uid=gradient_uid, ignore_existing=ignore_existing)
|