tensor-tang
|
6a7f83d45d
|
enable gru jitcode and refine act and lstm jitcode
test=develop
|
6 years ago |
tensor-tang
|
e3b61cf52b
|
init gru jitcode and fix lstm jitcode
test=develop
|
6 years ago |
tensor-tang
|
f913860873
|
jitkernel lstm refer support peephole
test=develop
|
6 years ago |
tensor-tang
|
2f9b5f2383
|
Merge branch 'develop' into fea/jit/rnn
|
6 years ago |
tensor-tang
|
10fb4ceefc
|
Merge pull request #14351 from tpatejko/tpatejko/mkldnn-elementwise_mul
[MKLDNN][JIT][AVX512] Elementwise Mul
|
6 years ago |
tensor-tang
|
c2cfb03a72
|
add lstm jitcode
|
6 years ago |
tensor-tang
|
4dbdfa60ef
|
sigmoid and tanh support all size
test=develop
|
6 years ago |
tensor-tang
|
ccb8963705
|
refine exp jitcode with all size
test=develop
|
6 years ago |
tensor-tang
|
d3eae8f61b
|
refine relu and fix addrelu test
|
6 years ago |
tensor-tang
|
ba3eaed7a7
|
exp support all size
|
6 years ago |
Michal Gallus
|
c69c41604e
|
MKLDNN elementwise_mul: Move Kernel to KernelPool to avoid segfaults
test=develop
|
6 years ago |
tensor-tang
|
f65ddff8d1
|
unify act jitcode of relu, exp, sigmoid and tanh
|
6 years ago |
tensor-tang
|
6a159071b6
|
add vtanh jitcode of size 8
|
6 years ago |
tensor-tang
|
046374bcd1
|
add vsigmoid jitcode of size 8
|
6 years ago |
tensor-tang
|
ee2a7f1b8c
|
refine exp and fix error on avx
test=develop
|
6 years ago |
tensor-tang
|
1e06a32a0d
|
add vexp jitcode of size 8
test=develop
|
6 years ago |
tensor-tang
|
0043c42b3e
|
add vrelu jitcode
test=develop
|
6 years ago |
tensor-tang
|
5e64244f25
|
add vaddbias jitcode
test=develop
|
7 years ago |
tensor-tang
|
3d950a812d
|
combine jitcode of vscal
|
7 years ago |
tensor-tang
|
03e11f3fc9
|
add vscal jitcode
|
7 years ago |
tensor-tang
|
382307b943
|
refine code
test=develop
|
7 years ago |
tensor-tang
|
b68ececb73
|
add vaddrelu jitcode
test=develop
|
7 years ago |
tensor-tang
|
bb09e31020
|
add vadd jitcode
test=develop
|
7 years ago |
tensor-tang
|
9255119fd9
|
refine jit vmul with all size
|
7 years ago |
tensor-tang
|
a9c1824131
|
refine jit vmul code supporting multiple of 2
|
7 years ago |
tensor-tang
|
a3377f7b0a
|
refine jitcode and add vmul jitcode implementation
|
7 years ago |