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,852 | adadelta_op.h |
![]() |
- | rw-r--r-- | 5,120 | adadelta_op_gpu.cu |
![]() |
- | rw-r--r-- | 9,940 | adagrad_fused.cc |
![]() |
- | rw-r--r-- | 16,632 | adagrad_fused.h |
![]() |
- | rw-r--r-- | 51,128 | adagrad_fused_op_gpu.cu |
![]() |
- | rw-r--r-- | 7,970 | adagrad_fused_op_gpu.cuh |
![]() |
- | rw-r--r-- | 7,563 | adagrad_op.cc |
![]() |
- | rw-r--r-- | 16,945 | adagrad_op.h |
![]() |
- | rw-r--r-- | 7,611 | adagrad_op_gpu.cu |
![]() |
- | rw-r--r-- | 5,442 | adam_op.cc |
![]() |
- | rw-r--r-- | 22,038 | adam_op.h |
![]() |
- | rw-r--r-- | 12,625 | 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-- | 8,385 | fp16_momentum_sgd_op.cu |
![]() |
- | rw-r--r-- | 2,292 | fp16_momentum_sgd_op.h |
![]() |
- | rw-r--r-- | 3,911 | fp32_momentum_sgd_op.cu |
![]() |
- | rw-r--r-- | 2,015 | fp32_momentum_sgd_op.h |
![]() |
- | rw-r--r-- | 3,936 | 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,975 | iter_op.cc |
![]() |
- | rw-r--r-- | 3,323 | 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-- | 922 | lars_op_gpu.cu |
![]() |
- | rw-r--r-- | 1,621 | learning_rate_adaption_op.cc |
![]() |
- | rw-r--r-- | 1,974 | learning_rate_adaption_op.h |
![]() |
- | rw-r--r-- | 15,135 | learning_rate_functors.h |
![]() |
- | rw-r--r-- | 7,968 | learning_rate_op.cc |
![]() |
- | rw-r--r-- | 12,696 | learning_rate_op.h |
![]() |
- | rw-r--r-- | 195 | learning_rate_op_gpu.cc |
![]() |
- | rw-r--r-- | 766 | math_lp.cc |
![]() |
- | rw-r--r-- | 655 | math_lp.h |
![]() |
- | rw-r--r-- | 3,740 | momentum_sgd_op.cc |
![]() |
- | rw-r--r-- | 6,035 | momentum_sgd_op.h |
![]() |
- | rw-r--r-- | 4,306 | 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,173 | rmsprop_op_gpu.cu |
![]() |
- | rw-r--r-- | 11,532 | rowwise_adagrad_fused.cc |
![]() |
- | rw-r--r-- | 29,598 | rowwise_adagrad_fused.h |
![]() |
- | rw-r--r-- | 883 | rowwise_counter.cc |
![]() |
- | rw-r--r-- | 2,083 | rowwise_counter.h |
![]() |
- | rw-r--r-- | 2,834 | storm_op.cc |
![]() |
- | rw-r--r-- | 5,915 | 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,272 | wngrad_op.h |
![]() |
- | rw-r--r-- | 4,902 | yellowfin_op.cc |
![]() |
- | rw-r--r-- | 10,284 | yellowfin_op.h |
![]() |
- | rw-r--r-- | 2,491 | yellowfin_op_gpu.cu |