summaryrefslogtreecommitdiff
path: root/caffe2/sgd
ModeNameSize
-rw-r--r--CMakeLists.txt1789logplain
-rw-r--r--adadelta_op.cc2622logplain
-rw-r--r--adadelta_op.h5852logplain
-rw-r--r--adadelta_op_gpu.cu5120logplain
-rw-r--r--adagrad_op.cc4233logplain
-rw-r--r--adagrad_op.h10234logplain
-rw-r--r--adagrad_op_gpu.cu6763logplain
-rw-r--r--adam_op.cc4506logplain
-rw-r--r--adam_op.h16419logplain
-rw-r--r--adam_op_gpu.cu7161logplain
-rw-r--r--clip_tensor_op.cc1375logplain
-rw-r--r--clip_tensor_op.h1853logplain
-rw-r--r--fp16_momentum_sgd_op.cu8385logplain
-rw-r--r--fp16_momentum_sgd_op.h2292logplain
-rw-r--r--fp32_momentum_sgd_op.cu3911logplain
-rw-r--r--fp32_momentum_sgd_op.h2015logplain
-rw-r--r--ftrl_op.cc3936logplain
-rw-r--r--ftrl_op.h2219logplain
-rw-r--r--gftrl_op.cc2908logplain
-rw-r--r--gftrl_op.h1028logplain
-rw-r--r--iter_op.cc1942logplain
-rw-r--r--iter_op.h3325logplain
-rw-r--r--iter_op_gpu.cc227logplain
-rw-r--r--lars_op.cc1706logplain
-rw-r--r--lars_op.h2479logplain
-rw-r--r--lars_op_gpu.cu922logplain
-rw-r--r--learning_rate_adaption_op.cc1621logplain
-rw-r--r--learning_rate_adaption_op.h1974logplain
-rw-r--r--learning_rate_functors.h6199logplain
-rw-r--r--learning_rate_op.cc3288logplain
-rw-r--r--learning_rate_op.h6769logplain
-rw-r--r--learning_rate_op_gpu.cc195logplain
-rw-r--r--momentum_sgd_op.cc3740logplain
-rw-r--r--momentum_sgd_op.h5997logplain
-rw-r--r--momentum_sgd_op_gpu.cu4306logplain
-rw-r--r--rmsprop_op.cc1524logplain
-rw-r--r--rmsprop_op.h1980logplain
-rw-r--r--rmsprop_op_gpu.cu1173logplain
-rw-r--r--wngrad_op.cc2153logplain
-rw-r--r--wngrad_op.h7272logplain
-rw-r--r--yellowfin_op.cc4902logplain
-rw-r--r--yellowfin_op.h10284logplain
-rw-r--r--yellowfin_op_gpu.cu2491logplain