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.

dangqingqing 7abd1bdfff Fix cmake error when building with WITH_AVX=OFF.		7 years ago
..
detail	Fix cmake error when building with WITH_AVX=OFF.	7 years ago
CMakeLists.txt	Merge pull request #5310 from qingqing01/fix_max_pool	7 years ago
context_project.cc	follow comments	7 years ago
context_project.cu	follow comments	7 years ago
context_project.h	fix code format and doc	7 years ago
cross_entropy.cc	support sparse output for lookup table grad op (#5145)	7 years ago
cross_entropy.cu	support sparse output for lookup table grad op (#5145)	7 years ago
cross_entropy.h	refine CrossEntropyFunctor	7 years ago
gru_compute.cc	Add GRU Operator	7 years ago
gru_compute.cu	Fix End of Files in GRU Operator	7 years ago
gru_compute.h	Add TODO in the GRU Operator related	7 years ago
im2col.cc	follow comments	7 years ago
im2col.cu	add padding up, down, left, right	7 years ago
im2col.h	fix im2col interface	7 years ago
im2col_test.cc	follow comments	7 years ago
lstm_compute.cc	Several Enhancement	7 years ago
lstm_compute.cu	Several Enhancement	7 years ago
lstm_compute.h	Several Enhancement	7 years ago
math_function.cc	Add gradient check unit testing and fix bug.	7 years ago
math_function.cu	Add gradient check unit testing and fix bug.	7 years ago
math_function.h	Add gradient check unit testing and fix bug.	7 years ago
math_function_test.cc	Add unit testing for gemv and fix the gradien check for bais.	7 years ago
math_function_test.cu	Add unit testing for gemv and fix the gradien check for bais.	7 years ago
matmul.h	MatMul operator (#4856)	7 years ago
pooling.cc	Add code comments	7 years ago
pooling.cu	Add code comments	7 years ago
pooling.h	follow coments	7 years ago
selected_rows_functor.cc	add sparse support for sum op (#5093)	7 years ago
selected_rows_functor.cu	add sparse support for sum op (#5093)	7 years ago
selected_rows_functor.h	add sparse support for sum op (#5093)	7 years ago
selected_rows_functor_test.cc	add sparse support for sum op (#5093)	7 years ago
selected_rows_functor_test.cu	add sparse support for sum op (#5093)	7 years ago
sequence2batch.cc	Update lstm comments and fix bug.	7 years ago
sequence2batch.cu	Several Enhancement	7 years ago
sequence2batch.h	Add GRU Operator	7 years ago
sequence_pooling.cc	Resolve conflicts.	7 years ago
sequence_pooling.cu	Resolve conflicts.	7 years ago
sequence_pooling.h	Refine sequence max-pooling and add unit testing of gradient check.	7 years ago
softmax.cc	Add SoftmaxGradFunctor, and use SoftmaxGradFunctor in softmax_op instead.	7 years ago
softmax.cu	Add SoftmaxGradFunctor, and use SoftmaxGradFunctor in softmax_op instead.	7 years ago
softmax.h	Merge branch 'develop' into core_add_sequence_softmax_op	7 years ago
vol2col.cc	follow comments	7 years ago
vol2col.cu	Add vol2col functor and unit test	7 years ago
vol2col.h	Add vol2col functor and unit test	7 years ago
vol2col_test.cc	Remove template parameter for Tensor methods (#4937)	7 years ago