package info
(click to toggle)
Folder: sgd
![]() |
.. (parent) | |||
![]() |
- | rw-r--r-- | 1,789 | CMakeLists.txt |
![]() |
- | rw-r--r-- | 2,622 | adadelta_op.cc |
![]() |
- | rw-r--r-- | 5,897 | adadelta_op.h |
![]() |
- | rw-r--r-- | 5,190 | adadelta_op_gpu.cu |
![]() |
- | rw-r--r-- | 9,940 | adagrad_fused.cc |
![]() |
- | rw-r--r-- | 16,586 | adagrad_fused.h |
![]() |
- | rw-r--r-- | 55,894 | adagrad_fused_op_gpu.cu |
![]() |
- | rw-r--r-- | 8,605 | adagrad_fused_op_gpu.cuh |
![]() |
- | rw-r--r-- | 9,501 | adagrad_op.cc |
![]() |
- | rw-r--r-- | 18,035 | adagrad_op.h |
![]() |
- | rw-r--r-- | 7,757 | adagrad_op_gpu.cu |
![]() |
- | rw-r--r-- | 7,216 | adam_op.cc |
![]() |
- | rw-r--r-- | 26,277 | adam_op.h |
![]() |
- | rw-r--r-- | 12,913 | adam_op_gpu.cu |
![]() |
- | rw-r--r-- | 1,377 | clip_tensor_op.cc |
![]() |
- | rw-r--r-- | 1,853 | clip_tensor_op.h |
![]() |
- | rw-r--r-- | 1,795 | decay_adagrad_op.cc |
![]() |
- | rw-r--r-- | 3,215 | decay_adagrad_op.h |
![]() |
- | rw-r--r-- | 8,437 | fp16_momentum_sgd_op.cu |
![]() |
- | rw-r--r-- | 2,292 | fp16_momentum_sgd_op.h |
![]() |
- | rw-r--r-- | 3,945 | fp32_momentum_sgd_op.cu |
![]() |
- | rw-r--r-- | 2,015 | fp32_momentum_sgd_op.h |
![]() |
- | rw-r--r-- | 3,948 | ftrl_op.cc |
![]() |
- | rw-r--r-- | 2,219 | ftrl_op.h |
![]() |
- | rw-r--r-- | 2,908 | gftrl_op.cc |
![]() |
- | rw-r--r-- | 1,028 | gftrl_op.h |
![]() |
- | rw-r--r-- | 1,961 | iter_op.cc |
![]() |
- | rw-r--r-- | 3,379 | iter_op.h |
![]() |
- | rw-r--r-- | 227 | iter_op_gpu.cc |
![]() |
- | rw-r--r-- | 1,706 | lars_op.cc |
![]() |
- | rw-r--r-- | 2,479 | lars_op.h |
![]() |
- | rw-r--r-- | 956 | lars_op_gpu.cu |
![]() |
- | rw-r--r-- | 1,621 | learning_rate_adaption_op.cc |
![]() |
- | rw-r--r-- | 1,981 | learning_rate_adaption_op.h |
![]() |
- | rw-r--r-- | 15,158 | learning_rate_functors.h |
![]() |
- | rw-r--r-- | 8,272 | learning_rate_op.cc |
![]() |
- | rw-r--r-- | 12,880 | learning_rate_op.h |
![]() |
- | rw-r--r-- | 195 | learning_rate_op_gpu.cc |
![]() |
- | rw-r--r-- | 845 | math_lp.cc |
![]() |
- | rw-r--r-- | 655 | math_lp.h |
![]() |
- | rw-r--r-- | 3,740 | momentum_sgd_op.cc |
![]() |
- | rw-r--r-- | 6,049 | momentum_sgd_op.h |
![]() |
- | rw-r--r-- | 4,412 | momentum_sgd_op_gpu.cu |
![]() |
- | rw-r--r-- | 1,524 | rmsprop_op.cc |
![]() |
- | rw-r--r-- | 1,980 | rmsprop_op.h |
![]() |
- | rw-r--r-- | 1,207 | rmsprop_op_gpu.cu |
![]() |
- | rw-r--r-- | 12,492 | rowwise_adagrad_fused.cc |
![]() |
- | rw-r--r-- | 31,526 | rowwise_adagrad_fused.h |
![]() |
- | rw-r--r-- | 883 | rowwise_counter.cc |
![]() |
- | rw-r--r-- | 2,090 | rowwise_counter.h |
![]() |
- | rw-r--r-- | 2,834 | storm_op.cc |
![]() |
- | rw-r--r-- | 5,950 | storm_op.h |
![]() |
- | rw-r--r-- | 1,795 | weight_scale_op.cc |
![]() |
- | rw-r--r-- | 2,552 | weight_scale_op.h |
![]() |
- | rw-r--r-- | 1,163 | weight_scale_op_gpu.cc |
![]() |
- | rw-r--r-- | 2,153 | wngrad_op.cc |
![]() |
- | rw-r--r-- | 7,335 | wngrad_op.h |
![]() |
- | rw-r--r-- | 4,902 | yellowfin_op.cc |
![]() |
- | rw-r--r-- | 10,292 | yellowfin_op.h |
![]() |
- | rw-r--r-- | 2,559 | yellowfin_op_gpu.cu |