Searched refs:gradient_uid (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_util.py | 143 gradient_uid="__unsupported__"): argument
167 with _maybe_colocate_with(y.op, gradient_uid, colocate_gradients_with_ops):
293 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
296 with ops._colocate_with_for_gradient(op, gradient_uid): # pylint: disable=protected-access
526 gradient_uid = ops.get_default_graph().unique_name("uid")
536 gradient_uid)
589 with _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops):
592 out_grads = _AggregatedGrads(grads, op, gradient_uid, loop_state,
682 gradient_uid,
842 def _MultiDeviceAddN(tensor_list, gradient_uid): argument
[all …]
|
| D | control_flow_ops.py | 742 def EnterGradientColocation(self, op, gradient_uid): argument
745 self._outer_context.EnterGradientColocation(op, gradient_uid)
747 def ExitGradientColocation(self, op, gradient_uid): argument
750 self._outer_context.ExitGradientColocation(op, gradient_uid)
|
| /external/tensorflow/tensorflow/python/tpu/ |
| D | tpu.py | 366 def EnterGradientColocation(self, op, gradient_uid): argument
376 if gradient_uid == "__unsupported__":
399 cluster = parts[0] + "." + gradient_uid
405 def ExitGradientColocation(self, op, gradient_uid): argument
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 4124 def _colocate_with_for_gradient(self, op, gradient_uid, argument
4127 if gradient_uid is not None and self._control_flow_context is not None:
4128 self._control_flow_context.EnterGradientColocation(op, gradient_uid)
4132 self._control_flow_context.ExitGradientColocation(op, gradient_uid)
5116 def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): argument
5133 op, gradient_uid=gradient_uid, ignore_existing=ignore_existing)
|