/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 62 momentum = lambda: 0.9 function
|
D | momentum.py | 46 def __init__(self, learning_rate, momentum, argument
|
D | rmsprop.py | 68 momentum=0.0, argument
|
/external/tensorflow/tensorflow/python/keras/optimizer_v2/ |
D | rmsprop_test.py | 63 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
|
D | gradient_descent.py | 105 momentum=0.0, argument
|
D | rmsprop.py | 101 momentum=0.0, argument
|
D | gradient_descent_test.py | 302 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | normalization_v2.py | 103 momentum=0.99, argument 305 momentum=0.99, argument
|
D | normalization.py | 168 momentum=0.99, argument 520 def _assign_moving_average(self, variable, value, momentum, inputs_size): argument
|
/external/tensorflow/tensorflow/core/protobuf/tpu/ |
D | optimization_parameters.proto | 153 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/ |
D | normalization.py | 126 momentum=0.99, argument 178 momentum=0.99, argument
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | momentum_test.py | 35 def _update_nesterov_momentum_numpy(self, var, accum, g, lr, momentum): argument
|
D | rmsprop_test.py | 41 momentum=0.0, argument
|
/external/tensorflow/tensorflow/python/keras/distribute/ |
D | test_example.py | 61 momentum=0.9, argument
|
D | minimize_loss_test.py | 246 def testTrainNetworkWithBatchNorm(self, distribution, optimizer_fn, momentum, argument
|
/external/tensorflow/tensorflow/python/distribute/ |
D | single_loss_example.py | 84 momentum=0.9, argument
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | horizontal_loop_fusion_test.cc | 323 HloInstruction* momentum = builder.AddInstruction( in TEST_F() local
|
/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/core/kernels/ |
D | training_ops.cc | 3072 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
|
D | training_ops_gpu.cu.cc | 300 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
|
D | training_ops_test.cc | 315 auto momentum = Scalar(g, 0.9); in RMSProp() local
|
/external/tensorflow/tensorflow/python/keras/ |
D | optimizer_v1.py | 180 def __init__(self, lr=0.01, momentum=0., decay=0., nesterov=False, **kwargs): argument
|
D | backend.py | 1905 def moving_average_update(x, value, momentum): argument
|