Commit Graph

21 Commits (1ab4101d6cc11e27c5ba35f2d5a3f50983435e91)

Author SHA1 Message Date
Yihua Xu b085ecc258 Avoid the string as the key of map to improve the jit performance (#21292)
5 years ago
Yihua Xu eec9c9cbe7 Fix jit tls issue (#21151)
5 years ago
dengkaipeng 51536f7f52 StrideASum. test=develop
6 years ago
dengkaipeng 93701dba50 add jit kernel for softmax axis. test=develop
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 a0c37662b9 enable sgd jitkernel refer code and test
6 years ago
tensor-tang 15da2f9a0d add embseqpool jitkernel refer code, test and benchmark
6 years ago
tensor-tang ba02ac4692
use mat attr and refine test (#15448)
6 years ago
tensor-tang 8117725852 add jit kernel hsum, hmax and softmax refer code
6 years ago
tensor-tang 09c5786e22 add square jitkernel
6 years ago
tensor-tang c5623c87a3 init jit matmul kernel
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 74292f414c enable eltwise nchw16c mul nc
6 years ago
tensor-tang 720b55cbcf enable crf decoding and layer norm refer code
6 years ago
tensor-tang 00d3afbcc9 add gru refer functions, test and benchmark
6 years ago
tensor-tang bf9302f950 add lstm, peephole refer and test
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