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.

taixiurong 52b05baca3 fix some bug in transformer training in xpu (#31918)		5 years ago
..
CMakeLists.txt	Compiling operator libraries with Unity build (#29130)	5 years ago
adadelta_op.cc	Polish some error message in opeators (#27876)	6 years ago
adadelta_op.cu	…
adadelta_op.h	Adadelta Optimizer (#26590)	6 years ago
adagrad_op.cc	enhance cvm bpr_loss adam adagrad adamax ftrl error message, test=develop (#24452)	6 years ago
adagrad_op.cu	…
adagrad_op.h	enhance cvm bpr_loss adam adagrad adamax ftrl error message, test=develop (#24452)	6 years ago
adam_op.cc	Support pure fp16 training for AMP API. (#29544)	5 years ago
adam_op.cu	Support pure fp16 training for AMP API. (#29544)	5 years ago
adam_op.h	Support pure fp16 training for AMP API. (#29544)	5 years ago
adam_op_xpu.cc	fix some bug in transformer training in xpu (#31918)	5 years ago
adamax_op.cc	enhance cvm bpr_loss adam adagrad adamax ftrl error message, test=develop (#24452)	6 years ago
adamax_op.cu	…
adamax_op.h	enhance cvm bpr_loss adam adagrad adamax ftrl error message, test=develop (#24452)	6 years ago
decayed_adagrad_op.cc	Enhance Op's Error Message (#27455)	6 years ago
decayed_adagrad_op.cu	…
decayed_adagrad_op.h	Enhance Op's Error Message (#27455)	6 years ago
dgc_momentum_op.cc	modify error report message, test=develop (#26743)	6 years ago
dgc_momentum_op.cu	…
dgc_momentum_op.h	…
dpsgd_op.cc	fix error message (#27318)	6 years ago
dpsgd_op.h	Polish some error message in opeators (#27876)	6 years ago
ftrl_op.cc	enhance error message, test=develop (#30220)	5 years ago
ftrl_op.cu	…
ftrl_op.h	FTRL with sparse update, test=develop (#22092)	6 years ago
lamb_op.cc	Modify the calculation logic of LambOptimizer (#29313)	5 years ago
lamb_op.cu	…
lamb_op.h	Modify the calculation logic of LambOptimizer (#29313)	5 years ago
lamb_op_xpu.cc	lamb_op_xpu;test=kunlun (#31012)	5 years ago
lars_momentum_op.cc	modified the implement of Lars optimizer (#26733)	6 years ago
lars_momentum_op.cu	modified the implement of Lars optimizer (#26733)	6 years ago
lars_momentum_op.h	Enhance Op's Error Message (#27455)	6 years ago
momentum_op.cc	Support pure fp16 training for AMP API. (#29544)	5 years ago
momentum_op.cu	…
momentum_op.h	Support pure fp16 training for AMP API. (#29544)	5 years ago
momentum_op_xpu.cc	Fix xpu error message (#28061)	6 years ago
proximal_adagrad_op.cc	Enhance error message of prefetch_op, proximal_adagrad_op, proximal_gd_op (#24436)	6 years ago
proximal_adagrad_op.cu	…
proximal_adagrad_op.h	Optimize compilation time with Unity Build (#29733)	5 years ago
proximal_gd_op.cc	Enhance error message of prefetch_op, proximal_adagrad_op, proximal_gd_op (#24436)	6 years ago
proximal_gd_op.cu	…
proximal_gd_op.h	Optimize compilation time with Unity Build (#29733)	5 years ago
rmsprop_op.cc	fix error message (#27318)	6 years ago
rmsprop_op.cu	fix adam (#27343)	6 years ago
rmsprop_op.h	Optimize compilation time with Unity Build (#29733)	5 years ago
rmsprop_op_xpu.cc	add rmsprop_op_xpu test=kunlun (#30493)	5 years ago
sgd_op.cc	fix error message (#27318)	6 years ago
sgd_op.cu	fix error message (#27318)	6 years ago
sgd_op.h	fix error message (#27318)	6 years ago
sgd_op_xpu.cc	Fix xpu error message (#28061)	6 years ago
unity_build_rule.cmake	Optimize compilation time with Unity Build (#29733)	5 years ago