Searched defs:colocate_gradients_with_ops (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 44 colocate_gradients_with_ops=False, argument
376 colocate_gradients_with_ops=False, argument
|
| D | gradients_util.py | 70 def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops, func_graphs, argument
140 colocate_gradients_with_ops, argument
283 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
472 colocate_gradients_with_ops=False, argument
|
| D | control_flow_state.py | 758 colocate_gradients_with_ops): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 85 colocate_gradients_with_ops=False, argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 432 colocate_gradients_with_ops=False, name=None, argument
493 colocate_gradients_with_ops=False, argument
|