Commit Graph

40 Commits (310edc0d0c9050f1f01c108655493c1935c00214)

Author SHA1 Message Date
dengkaipeng 93701dba50 add jit kernel for softmax axis. test=develop
6 years ago
tensor-tang 45bdd84dac enhance the jitkernel helper and add unit tests
6 years ago
tensor-tang 14a764c930 simplify the jitkernel templates and tests
6 years ago
tensor-tang 802f362ac4 unify the kernelfuncs cache and add unit test
6 years ago
tensor-tang 641b3cccce add vbroadcast mkl code and jitcode
6 years ago
tensor-tang 41a1270856 add vbroadcast jitkernel refer code and use it
6 years ago
tensor-tang 867e93b21a add jitkernel vcopy and speedup unit test time
6 years ago
tensor-tang 8e04133719 add benchmark and mkl sgd implement
6 years ago
tensor-tang 5aea2cd2e0
Merge pull request #15652 from tensor-tang/refine/pyramiddnn
6 years ago
Yihua Xu 685a20ef56 Add JIT CRF_decoding and Layer_norm unit-test (#15699)
6 years ago
tensor-tang 75fc792d40 fix when table width larger than 64
6 years ago
tensor-tang 15da2f9a0d add embseqpool jitkernel refer code, test and benchmark
6 years ago
tensor-tang a6a1a92ef7
Merge pull request #15586 from tensor-tang/jit/cache
6 years ago
tensor-tang ba02ac4692
use mat attr and refine test (#15448)
6 years ago
tensor-tang 2b0811c3fb refine vadd jitkernel choice
6 years ago
tensor-tang d59f733551 refine softmax and use with cache
6 years ago
tensor-tang 8117725852 add jit kernel hsum, hmax and softmax refer code
6 years ago
tensor-tang b67584a6e9 jit benchmark use tensor
6 years ago
tensor-tang 316e44b1b7 fix unused warnings
6 years ago
tensor-tang 579d758254 fix jitkernel tests and refine benchmark
6 years ago
tensor-tang 09c5786e22 add square jitkernel
6 years ago
tensor-tang 266a5d2f52 implement matmul refer and mkl kernel
6 years ago
tensor-tang c5623c87a3 init jit matmul kernel
6 years ago
tensor-tang c3a9f3c4b2 fix typo and refine
6 years ago
tensor-tang 123b98f417 refine heigth and codesize and support all pool
6 years ago
tensor-tang 0145f40f45 use height from params of jitcode
6 years ago
tensor-tang e0591deebc enhance seqpool jitcode
6 years ago
tensor-tang 142bb41748 add seqpool jitkernel test and benchmark
6 years ago
tensor-tang 1aaec571c2 fix enum style
6 years ago
tensor-tang facfecbdd0 follow comment: reuse time function and change to upper case
6 years ago
tensor-tang d53c4756ad clean code and remove unused files
6 years ago
tensor-tang 77907a3502 refine benchmark template
6 years ago
tensor-tang 00d3afbcc9 add gru refer functions, test and benchmark
6 years ago
tensor-tang 6eec461725 add lstm peephole benchmark
6 years ago
tensor-tang bf951fa737 add refer vrelu, videntity, vexp, vsigmoid, vtanh and test and benchmark
6 years ago
tensor-tang e9216e82f9 add refer vscal, vaddbias and test and benchmark
6 years ago
tensor-tang 417d031f90 add refer vadd, vaddrelu, vsub and tests and benchmark
6 years ago
tensor-tang f3250097bc fix bug and mac compile
6 years ago
tensor-tang bc0df6a948 make typename tuples
6 years ago
tensor-tang 194ce2e92c add benchmark
6 years ago