Home	Sort by relevance Sort by last modified time

Full Search Definition Symbol File Path History   | | Help

    Searched refs:sign_decay (Results 1 - 10 of 10) sorted by null

/external/tensorflow/tensorflow/contrib/opt/python/training/
sign_decay_test.py	15 """Tests for sign_decay.""" 23 from tensorflow.contrib.opt.python.training import sign_decay 66 linear_decay_fn = sign_decay.get_linear_decay_fn(num_training_steps) 76 cosine_decay_fn = sign_decay.get_cosine_decay_fn(num_training_steps) 77 cosine_decay_2_fn = sign_decay.get_cosine_decay_fn( 93 restart_decay_fn = sign_decay.get_restart_decay_fn(num_training_steps) 94 restart_decay_2_fn = sign_decay.get_restart_decay_fn(
addsign_test.py	24 from tensorflow.contrib.opt.python.training import sign_decay 150 sign_decay_fn = sign_decay.get_linear_decay_fn(decay_steps) 250 sign_decay_fn = sign_decay.get_linear_decay_fn(decay_steps)
powersign_test.py	25 from tensorflow.contrib.opt.python.training import sign_decay 152 sign_decay_fn = sign_decay.get_linear_decay_fn(decay_steps) 255 sign_decay_fn = sign_decay.get_linear_decay_fn(decay_steps)
/external/tensorflow/tensorflow/compiler/tests/
addsign_test.py	25 from tensorflow.contrib.opt.python.training import sign_decay 133 sign_decay_fn = sign_decay.get_linear_decay_fn(decay_steps)
powersign_test.py	26 from tensorflow.contrib.opt.python.training import sign_decay 133 sign_decay_fn = sign_decay.get_linear_decay_fn(decay_steps)
/external/tensorflow/tensorflow/core/kernels/
training_ops_test.cc	254 auto sign_decay = Scalar(g, 0.9); local 258 {var, m, lr, alpha, sign_decay, beta, grad}); 291 auto sign_decay = Scalar(g, 0.9); local 295 {var, m, lr, logbase, sign_decay, beta, grad});
training_ops.h	206 typename TTypes<T>::ConstScalar sign_decay, 217 typename TTypes<T>::ConstScalar sign_decay,
training_ops_gpu.cu.cc	282 typename TTypes<T>::ConstScalar sign_decay, 298 // var.device(d) -= lr() * (alpha() + sign_decay() * sign_gm) * grad; 302 auto sign_decay_bcast = sign_decay.reshape(single).broadcast(bcast); 314 typename TTypes<T>::ConstScalar sign_decay, 330 // auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); 335 auto sign_decay_bcast = sign_decay.reshape(single).broadcast(bcast);
training_ops.cc	433 typename TTypes<T>::ConstScalar sign_decay, 438 var.device(d) -= lr() * (alpha() + sign_decay() * sign_gm) * grad; 448 typename TTypes<T>::ConstScalar sign_decay, 453 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp(); 3913 const Tensor& sign_decay = ctx->input(4); variable 4020 const Tensor& sign_decay = ctx->input(4); variable     [all...]
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
training_ops.cc	900 xla::XlaOp sign_decay = ctx->Input(4); variable     [all...]

Completed in 1138 milliseconds