OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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