| .. |
|
adadelta_op_gpu.cu
|
|
|
|
adadelta_op.cc
|
|
|
|
adadelta_op.h
|
use irange for loops (#70248)
|
2022-01-06 23:14:29 -08:00 |
|
adagrad_fused_op_gpu.cu
|
Fix unused variable warnings (#72410)
|
2022-02-07 18:25:29 +00:00 |
|
adagrad_fused_op_gpu.cuh
|
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
|
2021-09-29 09:55:43 -07:00 |
|
adagrad_fused.cc
|
|
|
|
adagrad_fused.h
|
use irange for loops (#70248)
|
2022-01-06 23:14:29 -08:00 |
|
adagrad_op_gpu.cu
|
Refactor cub namespace handling (#66219)
|
2021-10-25 14:37:09 -07:00 |
|
adagrad_op.cc
|
Fix bytes_written and bytes_read (#64244)
|
2021-09-01 13:35:41 -07:00 |
|
adagrad_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
adam_op_gpu.cu
|
Refactor cub namespace handling (#66219)
|
2021-10-25 14:37:09 -07:00 |
|
adam_op.cc
|
|
|
|
adam_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
clip_tensor_op.cc
|
|
|
|
clip_tensor_op.h
|
|
|
|
CMakeLists.txt
|
|
|
|
decay_adagrad_op.cc
|
|
|
|
decay_adagrad_op.h
|
|
|
|
fp16_momentum_sgd_op.cu
|
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
|
2021-09-29 09:55:43 -07:00 |
|
fp16_momentum_sgd_op.h
|
|
|
|
fp32_momentum_sgd_op.cu
|
|
|
|
fp32_momentum_sgd_op.h
|
|
|
|
ftrl_op.cc
|
|
|
|
ftrl_op.h
|
|
|
|
gftrl_op.cc
|
|
|
|
gftrl_op.h
|
|
|
|
iter_op_gpu.cc
|
|
|
|
iter_op.cc
|
|
|
|
iter_op.h
|
|
|
|
lars_op_gpu.cu
|
|
|
|
lars_op.cc
|
|
|
|
lars_op.h
|
|
|
|
learning_rate_adaption_op.cc
|
|
|
|
learning_rate_adaption_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
learning_rate_functors.h
|
Fix some typos.
|
2022-04-11 21:55:59 +00:00 |
|
learning_rate_op_gpu.cc
|
|
|
|
learning_rate_op.cc
|
|
|
|
learning_rate_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
math_lp.cc
|
|
|
|
math_lp.h
|
|
|
|
momentum_sgd_op_gpu.cu
|
|
|
|
momentum_sgd_op.cc
|
|
|
|
momentum_sgd_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
rmsprop_op_gpu.cu
|
|
|
|
rmsprop_op.cc
|
|
|
|
rmsprop_op.h
|
|
|
|
rowwise_adagrad_fused.cc
|
support counter-based fused rowwise adagrad (#66177)
|
2021-10-06 13:50:43 -07:00 |
|
rowwise_adagrad_fused.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
rowwise_counter.cc
|
|
|
|
rowwise_counter.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
storm_op.cc
|
|
|
|
storm_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
weight_scale_op_gpu.cc
|
|
|
|
weight_scale_op.cc
|
|
|
|
weight_scale_op.h
|
|
|
|
wngrad_op.cc
|
|
|
|
wngrad_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
|
yellowfin_op_gpu.cu
|
|
|
|
yellowfin_op.cc
|
|
|
|
yellowfin_op.h
|
use irange for loops (#69533)
|
2021-12-07 16:53:27 -08:00 |
