HomeSort by relevance Sort by last modified time
    Searched refs:logbase (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
training_ops_test.cc 290 auto logbase = Scalar(g, 2); local
295 {var, m, lr, logbase, sign_decay, beta, grad});
training_ops_gpu.cu.cc 231 typename TTypes<T>::ConstScalar logbase,
248 // auto grad_scale = (logbase() * sign_decay() * sign_gm).exp();
252 auto logbase_bcast = logbase.reshape(single).broadcast(bcast);
training_ops.h 181 typename TTypes<T>::ConstScalar logbase,
training_ops.cc 385 typename TTypes<T>::ConstScalar logbase,
391 auto grad_scale = (logbase() * sign_decay() * sign_gm).exp();
3413 const Tensor& logbase = ctx->input(3); variable
    [all...]

Completed in 177 milliseconds