Commit Graph

31 Commits (41b59555387616edef6bd5ef1b9093ab92b90db1)

Author SHA1 Message Date
dengkaipeng eb2123e12d fix doc and jit. test=develop
6 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 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 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 0145f40f45 use height from params of jitcode
6 years ago
tensor-tang 142bb41748 add seqpool jitkernel test and benchmark
6 years ago
tensor-tang 3e01a4048f add refer seqpool jitkernel
6 years ago
tensor-tang 1aaec571c2 fix enum style
6 years ago
tensor-tang d53c4756ad clean code and remove unused files
6 years ago
tensor-tang c187a7c618 add more impls of lstm and gru and fix build on win
6 years ago
tensor-tang 83d075aa79 fix lstm and gru jitcode
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 5e97be7ba7 enable jitkernel mkl vexp, vsigmoid and vtanh
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
tensor-tang bc0df6a948 make typename tuples
6 years ago
tensor-tang d4cab7d948 use jitkernel in one file
6 years ago
tensor-tang 53709e7e61 refine names
6 years ago