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.

chengduo 7453857324 Make fuse_all_reduce_op_pass support mix_precision (#17652)		6 years ago
..
CMakeLists.txt	Refine operator cmake (#14413)	6 years ago
adadelta_op.cc	Change tensor uses proto::VarType::type	6 years ago
adadelta_op.cu	Fix Eigen macro when using GPU	6 years ago
adadelta_op.h	rewrite variable type	6 years ago
adagrad_op.cc	Change tensor uses proto::VarType::type	6 years ago
adagrad_op.cu	Fix Eigen macro when using GPU	6 years ago
adagrad_op.h	rewrite variable type	6 years ago
adam_op.cc	Add LAMB Optimizer support (#17489)	6 years ago
adam_op.cu	Fix Eigen macro when using GPU	6 years ago
adam_op.h	Add LAMB Optimizer support (#17489)	6 years ago
adamax_op.cc	Change tensor uses proto::VarType::type	6 years ago
adamax_op.cu	Fix Eigen macro when using GPU	6 years ago
adamax_op.h	rewrite variable type	6 years ago
decayed_adagrad_op.cc	Change tensor uses proto::VarType::type	6 years ago
decayed_adagrad_op.cu	Fix Eigen macro when using GPU	6 years ago
decayed_adagrad_op.h	rewrite variable type	6 years ago
ftrl_op.cc	Change tensor uses proto::VarType::type	6 years ago
ftrl_op.cu	Fix Eigen macro when using GPU	6 years ago
ftrl_op.h	rewrite variable type	6 years ago
lamb_op.cc	Update lamb optimizer (#18333)	6 years ago
lamb_op.cu	Add LAMB Optimizer support (#17489)	6 years ago
lamb_op.h	Update lamb optimizer (#18333)	6 years ago
lars_momentum_op.cc	Polish code style	6 years ago
lars_momentum_op.cu	Refine operator cmake (#14413)	6 years ago
lars_momentum_op.h	Refine operator cmake (#14413)	6 years ago
momentum_op.cc	Polish code style	6 years ago
momentum_op.cu	Fp16 training (#14992)	6 years ago
momentum_op.h	fix const_cast	6 years ago
proximal_adagrad_op.cc	Change tensor uses proto::VarType::type	6 years ago
proximal_adagrad_op.cu	Fix Eigen macro when using GPU	6 years ago
proximal_adagrad_op.h	Refine operator cmake (#14413)	6 years ago
proximal_gd_op.cc	Change tensor uses proto::VarType::type	6 years ago
proximal_gd_op.cu	Fix Eigen macro when using GPU	6 years ago
proximal_gd_op.h	Refine operator cmake (#14413)	6 years ago
rmsprop_op.cc	Refine operator cmake (#14413)	6 years ago
rmsprop_op.cu	Fix Eigen macro when using GPU	6 years ago
rmsprop_op.h	fix const_cast	6 years ago
sgd_op.cc	Make fuse_all_reduce_op_pass support mix_precision (#17652)	6 years ago
sgd_op.cu	Make fuse_all_reduce_op_pass support mix_precision (#17652)	6 years ago
sgd_op.h	simplify the jitkernel templates and tests	6 years ago