Searched defs:colocate_gradients_with_ops (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_impl.py | 48 colocate_gradients_with_ops=False, argument
380 colocate_gradients_with_ops=False, argument
|
| D | gradients_util.py | 73 def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops, func_graphs, argument
143 colocate_gradients_with_ops, argument
294 def _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops): # pylint: disable=invalid… argument
484 colocate_gradients_with_ops=False, argument
|
| D | control_flow_state.py | 762 colocate_gradients_with_ops): argument
|
| /external/tensorflow/tensorflow/python/training/experimental/ |
| D | loss_scale_optimizer.py | 89 colocate_gradients_with_ops=False, argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 365 colocate_gradients_with_ops=False, name=None, argument
426 colocate_gradients_with_ops=False, argument
|