Commit Graph

15 Commits (54474637ae82a97fe23ae607d6aaae0ad05fc2fa)

Author SHA1 Message Date
dengkaipeng eb2123e12d fix doc and jit. test=develop
6 years ago
dengkaipeng 90bd038d35 fix format. test=develop
6 years ago
dengkaipeng f45aced59b add jit test. develop=test
7 years ago
dengkaipeng 51536f7f52 StrideASum. test=develop
7 years ago
dengkaipeng 93701dba50 add jit kernel for softmax axis. test=develop
7 years ago
tensor-tang 45bdd84dac enhance the jitkernel helper and add unit tests
7 years ago
tensor-tang 14a764c930 simplify the jitkernel templates and tests
7 years ago
tensor-tang 6e1ee7fb57 cache softmax kernel func
7 years ago
tensor-tang d59f733551 refine softmax and use with cache
7 years ago
tensor-tang 7383eefd2d add softmax mix and mkl code
7 years ago
tensor-tang 1aaec571c2 fix enum style
7 years ago
tensor-tang a369c8025b fix typo of mix impl of lstm
7 years ago
tensor-tang d53c4756ad clean code and remove unused files
7 years ago
tensor-tang c187a7c618 add more impls of lstm and gru and fix build on win
7 years ago
tensor-tang f332f589bc add more impls of sigmoid and vtanh
7 years ago