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

  /external/tensorflow/tensorflow/core/kernels/
training_ops_gpu.cu.cc 82 struct ApplyMomentum<GPUDevice, T> {
273 template struct functor::ApplyMomentum<GPUDevice, Eigen::half>;
274 template struct functor::ApplyMomentum<GPUDevice, float>;
275 template struct functor::ApplyMomentum<GPUDevice, double>;
training_ops.h 121 struct ApplyMomentum {
training_ops.cc 268 struct ApplyMomentum<CPUDevice, T> {
    [all...]

Completed in 60 milliseconds