| .. |
|
adadelta_op.cc
|
Add Adadelta optimizer to caffe2 (#9088)
|
2018-07-24 20:09:21 -07:00 |
|
adadelta_op.h
|
Add Adadelta optimizer to caffe2 (#9088)
|
2018-07-24 20:09:21 -07:00 |
|
adadelta_op_gpu.cu
|
Add Adadelta optimizer to caffe2 (#9088)
|
2018-07-24 20:09:21 -07:00 |
|
adagrad_op.cc
|
Add cost inference of fwd sparse operators and sparse adagrad (#9314)
|
2018-08-06 08:39:16 -07:00 |
|
adagrad_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
adagrad_op_gpu.cu
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
adam_op.cc
|
Revert D9554375: Support lr adaption for SparseAdam and RowWiseSparseAdam
|
2018-08-31 07:54:31 -07:00 |
|
adam_op.h
|
Revert D9554375: Support lr adaption for SparseAdam and RowWiseSparseAdam
|
2018-08-31 07:54:31 -07:00 |
|
adam_op_gpu.cu
|
Remove template parameter from Tensor (#9939)
|
2018-07-27 10:56:39 -07:00 |
|
clip_tensor_op.cc
|
|
|
|
clip_tensor_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
CMakeLists.txt
|
HIP Operators Generator--> HipOpG (#9322)
|
2018-07-19 00:26:06 -07:00 |
|
fp16_momentum_sgd_op.cu
|
HIP Operators Generator--> HipOpG (#9322)
|
2018-07-19 00:26:06 -07:00 |
|
fp16_momentum_sgd_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
fp32_momentum_sgd_op.cu
|
HIP Operators Generator--> HipOpG (#9322)
|
2018-07-19 00:26:06 -07:00 |
|
fp32_momentum_sgd_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
ftrl_op.cc
|
|
|
|
ftrl_op.h
|
|
|
|
gftrl_op.cc
|
Add FTRL Optimzier with Group Lasso regularizer (#9074)
|
2018-07-06 13:41:00 -07:00 |
|
gftrl_op.h
|
Add FTRL Optimzier with Group Lasso regularizer (#9074)
|
2018-07-06 13:41:00 -07:00 |
|
iter_op.cc
|
|
|
|
iter_op.h
|
Remove template parameter from Tensor (#9939)
|
2018-07-27 10:56:39 -07:00 |
|
iter_op_gpu.cc
|
|
|
|
lars_op.cc
|
Add CUDA implementation of LARS --caffe2 (#10509)
|
2018-08-23 12:55:57 -07:00 |
|
lars_op.h
|
Add CUDA implementation of LARS --caffe2 (#10509)
|
2018-08-23 12:55:57 -07:00 |
|
lars_op_gpu.cu
|
Add CUDA implementation of LARS --caffe2 (#10509)
|
2018-08-23 12:55:57 -07:00 |
|
learning_rate_adaption_op.cc
|
Update from facebook (#7696)
|
2018-05-19 23:10:48 -07:00 |
|
learning_rate_adaption_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
learning_rate_functors.h
|
|
|
|
learning_rate_op.cc
|
|
|
|
learning_rate_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
learning_rate_op_gpu.cc
|
|
|
|
momentum_sgd_op.cc
|
|
|
|
momentum_sgd_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
momentum_sgd_op_gpu.cu
|
HIP Operators Generator--> HipOpG (#9322)
|
2018-07-19 00:26:06 -07:00 |
|
rmsprop_op.cc
|
Fix Eigen issue on OS X with CUDA and nvcc compile (#9350)
|
2018-07-11 14:00:05 -07:00 |
|
rmsprop_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
rmsprop_op_gpu.cu
|
HIP Operators Generator--> HipOpG (#9322)
|
2018-07-19 00:26:06 -07:00 |
|
wngrad_op.cc
|
Implementation and operator test for Wngrad optimizer (#8999)
|
2018-07-13 18:11:41 -07:00 |
|
wngrad_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
yellowfin_op.cc
|
|
|
|
yellowfin_op.h
|
Workaround for Cuda9.2 and GCC7 compilation errors (#10510)
|
2018-08-14 20:54:52 -07:00 |
|
yellowfin_op_gpu.cu
|
[caffe2] YellowFin parameter update GPU code fix. (#6993)
|
2018-05-30 16:36:08 -07:00 |
3073051a18