Home
last modified time | relevance | path

Searched refs:colocate_gradients_with_ops (Results 1 – 25 of 29) sorted by relevance

12

/external/tensorflow/tensorflow/python/ops/
Dgradients_impl.py48 colocate_gradients_with_ops=False, argument
170 ys, xs, grad_ys, name, colocate_gradients_with_ops,
380 colocate_gradients_with_ops=False, argument
410 "colocate_gradients_with_ops": colocate_gradients_with_ops,
477 colocate_gradients_with_ops=True,
Dgradients_util.py73 def _PendingCount(to_ops, from_ops, colocate_gradients_with_ops, func_graphs, argument
125 between_op_list, between_ops, colocate_gradients_with_ops)
143 colocate_gradients_with_ops, 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
296 if colocate_gradients_with_ops:
484 colocate_gradients_with_ops=False, argument
536 grad_ys = _DefaultGradYs(grad_ys, ys, colocate_gradients_with_ops,
552 to_ops, from_ops, colocate_gradients_with_ops, func_graphs, xs_set)
590 with _maybe_colocate_with(op, gradient_uid, colocate_gradients_with_ops):
Dimage_grad_deterministic_test.py122 colocate_gradients_with_ops=True)[0]
Dnccl_ops_test.py125 reduce_tensors, inputs, losses, colocate_gradients_with_ops=True)
Dgradients_test.py97 gw = gradients.gradients(wx, [w], colocate_gradients_with_ops=True)[0]
111 gw1 = gradients.gradients(z, [w], colocate_gradients_with_ops=True)[0]
114 gw2 = gradients.gradients(z, [w], colocate_gradients_with_ops=False)[0]
130 gw1 = gradients.gradients(z, [w], colocate_gradients_with_ops=True)[0]
133 gw2 = gradients.gradients(z, [w], colocate_gradients_with_ops=False)[0]
147 gz_x = gradients.gradients(z, [x], colocate_gradients_with_ops=True,
Dcontrol_flow_state.py762 colocate_gradients_with_ops): argument
776 if colocate_gradients_with_ops:
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer.py89 colocate_gradients_with_ops=False, argument
124 colocate_gradients_with_ops=colocate_gradients_with_ops,
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-optimizer.pbtxt28 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
44 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-gradient-descent-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-proximal-gradient-descent-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-adadelta-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-adagrad-d-a-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-adam-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-momentum-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.mixed_precision.-mixed-precision-loss-scale-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-proximal-adagrad-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-adagrad-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-r-m-s-prop-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.experimental.-mixed-precision-loss-scale-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-ftrl-optimizer.pbtxt29 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'gr…
45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.tpu.-cross-shard-optimizer.pbtxt45 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
Dtensorflow.train.-sync-replicas-optimizer.pbtxt57 …', \'var_list\', \'gate_gradients\', \'aggregation_method\', \'colocate_gradients_with_ops\', \'na…
/external/tensorflow/tensorflow/python/training/
Doptimizer.py365 colocate_gradients_with_ops=False, name=None, argument
410 colocate_gradients_with_ops=colocate_gradients_with_ops,
426 colocate_gradients_with_ops=False, argument
520 colocate_gradients_with_ops=colocate_gradients_with_ops)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_outside_compilation_test.py454 [c], [x], colocate_gradients_with_ops=True)[0]
478 [c], [x], colocate_gradients_with_ops=True)[0]
480 [d], [x], colocate_gradients_with_ops=True)[0]
498 colocate_gradients_with_ops=True)[0]
/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_deterministic_test.py131 colocate_gradients_with_ops=True)[0]

12