Home
last modified time | relevance | path

Searched defs:momentum (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Drmsprop_test.py61 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
Dmomentum_test.py40 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
62 momentum = lambda: 0.9 function
Dmomentum.py46 def __init__(self, learning_rate, momentum, argument
Drmsprop.py68 momentum=0.0, argument
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Drmsprop_test.py63 def _rmsprop_update_numpy(self, var, g, mg, rms, mom, lr, rho, momentum, argument
81 lr, rho, momentum, epsilon, centered): argument
471 momentum = lambda: 0.0 function
Dgradient_descent.py105 momentum=0.0, argument
Drmsprop.py101 momentum=0.0, argument
Dgradient_descent_test.py302 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_v2.py103 momentum=0.99, argument
305 momentum=0.99, argument
Dnormalization.py168 momentum=0.99, argument
520 def _assign_moving_average(self, variable, value, momentum, inputs_size): argument
/external/tensorflow/tensorflow/core/protobuf/tpu/
Doptimization_parameters.proto153 float momentum = 1; field
165 float momentum = 2; field
177 float momentum = 2; field
418 MomentumParameters momentum = 8; field
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dnormalization.py126 momentum=0.99, argument
178 momentum=0.99, argument
/external/tensorflow/tensorflow/compiler/tests/
Dmomentum_test.py35 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
Drmsprop_test.py41 momentum=0.0, argument
/external/tensorflow/tensorflow/python/keras/distribute/
Dtest_example.py61 momentum=0.9, argument
Dminimize_loss_test.py246 def testTrainNetworkWithBatchNorm(self, distribution, optimizer_fn, momentum, argument
/external/tensorflow/tensorflow/python/distribute/
Dsingle_loss_example.py84 momentum=0.9, argument
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion_test.cc323 HloInstruction* momentum = builder.AddInstruction( in TEST_F() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtraining_ops.cc154 xla::XlaOp momentum = ctx->Input(4); in Compile() local
212 xla::XlaOp momentum = ctx->Input(4); in Compile() local
688 xla::XlaOp momentum = ctx->Input("momentum"); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dtraining_ops.cc3072 const Tensor& momentum = ctx->input(4); in Compute() local
3196 const Tensor& momentum = ctx->input(5); in Compute() local
3301 const Tensor& momentum = ctx->input(4); in Compute() local
3423 const Tensor& momentum = ctx->input(5); in Compute() local
3965 const Tensor& momentum = ctx->input(5); in Compute() local
4055 const Tensor& momentum = ctx->input(6); in Compute() local
4214 const Tensor& momentum = ctx->input(5); in Compute() local
4347 const Tensor& momentum = ctx->input(6); in Compute() local
Dtraining_ops_gpu.cu.cc300 const T* momentum, bool use_nesterov, Tindex param_rows, in SparseApplyKerasMomentumKernel()
408 T momentum = pmomentum[0]; in ApplyRMSPropKernel() local
423 T momentum = pmomentum[0]; in ApplyCenteredRMSPropKernel() local
Dtraining_ops_test.cc315 auto momentum = Scalar(g, 0.9); in RMSProp() local
/external/tensorflow/tensorflow/python/keras/
Doptimizer_v1.py180 def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, **kwargs): argument
Dbackend.py1905 def moving_average_update(x, value, momentum): argument