| /external/tensorflow/tensorflow/contrib/eager/python/examples/l2hmc/ |
| D | l2hmc.py | 124 def _forward_lf(self, position, momentum, i): argument
147 def _backward_lf(self, position, momentum, i): argument
171 def _update_momentum_forward(self, position, momentum, t): argument
184 def _update_position_forward(self, position, momentum, t, mask, mask_inv): argument
197 def _update_momentum_backward(self, position, momentum, t): argument
210 def _update_position_backward(self, position, momentum, t, mask, mask_inv): argument
223 def _compute_accept_prob(self, position, momentum, position_post, argument
276 def hamiltonian(self, position, momentum): argument
|
| /external/tensorflow/tensorflow/python/training/ |
| D | rmsprop_test.py | 61 def _rmsprop_update_numpy(self, var, g, mg, rms, mom, lr, decay, momentum, argument
75 lr, decay, momentum, epsilon, centered): argument
473 momentum = lambda: 0.0 function
|
| D | momentum_test.py | 40 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
60 momentum = lambda: 0.9 function
|
| D | momentum.py | 46 def __init__(self, learning_rate, momentum, argument
|
| D | rmsprop.py | 64 momentum=0.0, argument
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | rmsprop_test.py | 60 def _rmsprop_update_numpy(self, var, g, mg, rms, mom, lr, rho, momentum, argument
78 lr, rho, momentum, epsilon, centered): argument
470 momentum = lambda: 0.0 function
|
| D | gradient_descent.py | 64 momentum=0.0, argument
|
| D | rmsprop.py | 66 momentum=0.0, argument
|
| D | gradient_descent_test.py | 293 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
|
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | momentum_test.py | 40 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
60 momentum = lambda: 0.9 function
|
| D | rmsprop_test.py | 57 def _rmsprop_update_numpy(self, var, g, mg, rms, mom, lr, decay, momentum, argument
71 lr, decay, momentum, centered): argument
|
| D | momentum.py | 42 def __init__(self, learning_rate, momentum, argument
|
| D | rmsprop.py | 61 momentum=0.0, argument
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | optimization_parameters.proto | 111 float momentum = 1; field
120 float momentum = 2; field
130 float momentum = 2; field
258 MomentumParameters momentum = 8; field
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | rmsprop_test.py | 41 momentum=0.0, argument
|
| D | momentum_test.py | 35 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | normalization.py | 109 momentum=0.99, argument
163 momentum=0.99, argument
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | single_loss_example.py | 91 momentum=0.9, argument
|
| D | minimize_loss_test.py | 209 def testTrainNetworkWithBatchNorm(self, distribution, optimizer_fn, momentum, argument
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| D | lars_optimizer.py | 50 momentum=0.9, argument
|
| D | weight_decay_optimizers_test.py | 49 def momentumw_update_numpy(param, g_t, m, lr=0.001, momentum=0.9, **_): argument
|
| D | weight_decay_optimizers.py | 304 def __init__(self, weight_decay, learning_rate, momentum, argument
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 138 momentum=0.99, argument
427 def _assign_moving_average(self, variable, value, momentum): argument
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| D | dynamic_rnn_estimator.py | 566 momentum=None, argument
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | training_ops.cc | 154 xla::XlaOp momentum = ctx->Input(4); in Compile() local
212 xla::XlaOp momentum = ctx->Input(4); in Compile() local
613 xla::XlaOp momentum = ctx->Input("momentum"); in Compile() local
|