Commit Graph

28 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 641b3cccce add vbroadcast mkl code and jitcode
7 years ago
tensor-tang 867e93b21a add jitkernel vcopy and speedup unit test time
7 years ago
tensor-tang 8e04133719 add benchmark and mkl sgd implement
7 years ago
tensor-tang a3a3d3d861 add embseqpool jitkernel mkl impl and use it
7 years ago
tensor-tang a6a1a92ef7
Merge pull request #15586 from tensor-tang/jit/cache
7 years ago
tensor-tang ba02ac4692
use mat attr and refine test (#15448)
7 years ago
tensor-tang 2b0811c3fb refine vadd jitkernel choice
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 09c5786e22 add square jitkernel
7 years ago
tensor-tang 266a5d2f52 implement matmul refer and mkl kernel
7 years ago
tensor-tang c5623c87a3 init jit matmul kernel
7 years ago
tensor-tang 92201d3956 support avg and sqrt pool and add mkl impl
7 years ago
tensor-tang 1aaec571c2 fix enum style
7 years ago
tensor-tang d53c4756ad clean code and remove unused files
7 years ago
tensor-tang 5e97be7ba7 enable jitkernel mkl vexp, vsigmoid and vtanh
7 years ago
tensor-tang ae17926987 enable jitkernel mkl vmul, vadd and vscal
7 years ago
tensor-tang 417d031f90 add refer vadd, vaddrelu, vsub and tests and benchmark
7 years ago
tensor-tang bc0df6a948 make typename tuples
7 years ago
tensor-tang d4cab7d948 use jitkernel in one file
7 years ago
tensor-tang 53709e7e61 refine names
7 years ago