Commit Graph

14 Commits (f086ebb8b9b8e21f7d0d2056f6a961da080edfb2)

Author SHA1 Message Date
QI JUN 61ec0b9516
Refine device context (#6433)
7 years ago
dangqingqing 2673657684 Move RowwiseAdd functor to math_funcion and Add ColwiseSum functor.
7 years ago
dangqingqing f5e367655e Use G++ to compile some cu operators.
7 years ago
dangqingqing d94c936bd5 Enhance unit testing.
7 years ago
dangqingqing 0c70bd28aa Enable initial hidden state and cell state in LSTM Operator.
7 years ago
dangqingqing bd680f157f fix compiling warning.
7 years ago
dangqingqing cd38286684 Add gradient check unit testing and fix bug.
7 years ago
dangqingqing 3d8b6ebcf8 Add LSTM backward implenmentation.
7 years ago
dangqingqing cf2608e383 update to the develop branch.
7 years ago
Yu Yang 65906ef1d0 Several Enhancement
7 years ago
dangqingqing d1fbf50b9e Add unit testing for forwad implementation.
7 years ago
dangqingqing 2a8dbd130d LSTM Operator forward implementation.
7 years ago
dangqingqing 3cace73701 Add lstm implementation.
7 years ago
dangqingqing 8728b3cce2 Add LSTM Operators.
7 years ago