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

  /external/tensorflow/tensorflow/core/kernels/
training_ops_gpu.cu.cc 103 struct ApplyAdam<GPUDevice, T> {
277 template struct functor::ApplyAdam<GPUDevice, Eigen::half>;
278 template struct functor::ApplyAdam<GPUDevice, float>;
279 template struct functor::ApplyAdam<GPUDevice, double>;
training_ops.h 130 struct ApplyAdam {
training_ops.cc 329 struct ApplyAdam<CPUDevice, T> : ApplyAdamNonCuda<CPUDevice, T> {};
    [all...]
  /external/tensorflow/tensorflow/core/profiler/g3doc/
profile_time.md 126 ApplyAdam 92.66ms (25.00%, 2.47%), 27
advise.md 72 top 2 graph node: train_step/update_seq2seq/output_projection/w/ApplyAdam, cpu: 84.52ms, accelerator: 0us, total: 84.52ms

Completed in 83 milliseconds