| /external/tensorflow/tensorflow/python/training/ |
| D | rmsprop_test.py | 57 def _rmsprop_update_numpy(self, var, g, mg, rms, mom, lr, decay, momentum, argument
71 lr, decay, momentum, epsilon, centered): argument
469 momentum = lambda: 0.0 function
|
| D | momentum_test.py | 36 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
58 momentum = lambda: 0.9 function
|
| D | momentum.py | 116 def __init__(self, learning_rate, momentum, argument
|
| D | rmsprop.py | 139 momentum=0.0, argument
|
| /external/tensorflow/tensorflow/core/protobuf/tpu/ |
| D | optimization_parameters.proto | 103 float momentum = 1; field
208 float momentum = 1; field
220 float momentum = 2; field
232 float momentum = 2; field
542 MomentumParameters momentum = 8; field
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | momentum_test.py | 31 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
|
| D | rmsprop_test.py | 37 momentum=0.0, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | single_loss_example.py | 80 momentum=0.9, argument
|
| /external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
| D | gradient_descent.py | 102 momentum=0.0, argument
|
| D | rmsprop.py | 98 momentum=0.0, 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
688 xla::XlaOp momentum = ctx->Input("momentum"); in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | horizontal_loop_fusion_test.cc | 328 HloInstruction* momentum = builder.AddInstruction( in TEST_F() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops.cc | 3104 const Tensor& momentum = ctx->input(4); in Compute() local
3220 const Tensor& momentum = ctx->input(5); in Compute() local
3325 const Tensor& momentum = ctx->input(4); in Compute() local
3439 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
4210 const Tensor& momentum = ctx->input(5); in Compute() local
4343 const Tensor& momentum = ctx->input(6); in Compute() local
|
| D | training_ops_gpu.cu.cc | 224 const T* momentum, bool use_nesterov, Tindex param_rows, in SparseApplyKerasMomentumKernel()
334 T momentum = pmomentum[0]; in ApplyRMSPropKernel() local
350 T momentum = pmomentum[0]; in ApplyCenteredRMSPropKernel() local
|
| D | training_ops_test.cc | 316 auto momentum = Scalar(g, 0.9); in RMSProp() local
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | optimizer_v1.py | 175 def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, **kwargs): argument
|
| D | backend.py | 1903 def moving_average_update(x, value, momentum): argument
|