You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Chengmo 4cbcc9b6da fix momentum op register (#29941) ... * fix momentum op register		4 years ago
..
CMakeLists.txt	Compiling operator libraries with Unity build (#29130)	4 years ago
adadelta_op.cc	Polish some error message in opeators (#27876)	4 years ago
adadelta_op.cu	…
adadelta_op.h	…
adagrad_op.cc	…
adagrad_op.cu	…
adagrad_op.h	…
adam_op.cc	…
adam_op.cu	Fix adam (#27778)	4 years ago
adam_op.h	xpu adam op (#28031)	4 years ago
adam_op_xpu.cc	fix crash in adam in xpu, *test=kunlun (#28433)	4 years ago
adamax_op.cc	…
adamax_op.cu	…
adamax_op.h	…
decayed_adagrad_op.cc	Enhance Op's Error Message (#27455)	4 years ago
decayed_adagrad_op.cu	…
decayed_adagrad_op.h	Enhance Op's Error Message (#27455)	4 years ago
dgc_momentum_op.cc	…
dgc_momentum_op.cu	…
dgc_momentum_op.h	…
dpsgd_op.cc	fix error message (#27318)	4 years ago
dpsgd_op.h	Polish some error message in opeators (#27876)	4 years ago
ftrl_op.cc	…
ftrl_op.cu	…
ftrl_op.h	…
lamb_op.cc	…
lamb_op.cu	…
lamb_op.h	…
lars_momentum_op.cc	…
lars_momentum_op.cu	…
lars_momentum_op.h	Enhance Op's Error Message (#27455)	4 years ago
momentum_op.cc	fix momentum op register (#29941)	4 years ago
momentum_op.cu	…
momentum_op.h	Remove some useless log. (#29300)	4 years ago
momentum_op_xpu.cc	Fix xpu error message (#28061)	4 years ago
proximal_adagrad_op.cc	…
proximal_adagrad_op.cu	…
proximal_adagrad_op.h	Optimize compilation time with Unity Build (#29733)	4 years ago
proximal_gd_op.cc	…
proximal_gd_op.cu	…
proximal_gd_op.h	Optimize compilation time with Unity Build (#29733)	4 years ago
rmsprop_op.cc	fix error message (#27318)	4 years ago
rmsprop_op.cu	…
rmsprop_op.h	Optimize compilation time with Unity Build (#29733)	4 years ago
sgd_op.cc	fix error message (#27318)	4 years ago
sgd_op.cu	fix error message (#27318)	4 years ago
sgd_op.h	fix error message (#27318)	4 years ago
sgd_op_xpu.cc	Fix xpu error message (#28061)	4 years ago
unity_build_rule.cmake	Optimize compilation time with Unity Build (#29733)	4 years ago