HomeSort by relevance Sort by last modified time
    Searched refs:sign_decay (Results 1 - 8 of 8) 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/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_gpu.cu.cc 200 typename TTypes<T>::ConstScalar sign_decay,
216 // var.device(d) -= lr() * (alpha() + sign_decay() * sign_gm) * grad;
220 auto sign_decay_bcast = sign_decay.reshape(single).broadcast(bcast);
232 typename TTypes<T>::ConstScalar sign_decay,
248 // auto grad_scale = (logbase() * sign_decay() * sign_gm).exp();
253 auto sign_decay_bcast = sign_decay.reshape(single).broadcast(bcast);
training_ops.h 171 typename TTypes<T>::ConstScalar sign_decay,
182 typename TTypes<T>::ConstScalar sign_decay,
training_ops.cc 371 typename TTypes<T>::ConstScalar sign_decay,
376 var.device(d) -= lr() * (alpha() + sign_decay() * sign_gm) * grad;
386 typename TTypes<T>::ConstScalar sign_decay,
391 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp();
3312 const Tensor& sign_decay = ctx->input(4); variable
3417 const Tensor& sign_decay = ctx->input(4); variable
    [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 466 milliseconds