nihui
|
aed05aa851
|
pnnx fuse more function to module (#4351)
* pnnx fuse more function to module
* rename some pass name
* fuse adjacent reshape, fuse pad conv2d
* fuse pad conv1d
|
3 years ago |
nihui
|
6967baaccc
|
pnnx convert torch bitwise left_shift right_shift (#4349)
|
3 years ago |
nihui
|
eceac35a7f
|
implement MultiheadAttention kdim vdim (#4347)
|
3 years ago |
nihui
|
498ca7341b
|
squeeze and expanddims 4d (#4346)
|
3 years ago |
Lry89757
|
6a47f8d15c
|
gridsample op support (#4288)
Co-authored-by: LRY89757 <LRY89757@users.noreply.github.com>
Co-authored-by: nihuini <nihuini@tencent.com>
Co-authored-by: nihui <shuizhuyuanluo@126.com>
|
3 years ago |
nihui
|
a2af6369d9
|
match inplace slice copy pattern, rewrite copy uses (#4338)
|
3 years ago |
nihui
|
5b28c1730e
|
implement ncnn fold and unfold (#4326)
|
3 years ago |
nihui
|
a12c24d328
|
pnnx convert fold unfold (#4325)
|
3 years ago |
nihui
|
b8d40a960f
|
pnnx convert nn.Softmax2d (#4324)
|
3 years ago |
nihui
|
bcf06bd1c0
|
fold new_full and full_like (#4323)
|
3 years ago |
nihui
|
6e49fa30dc
|
groupnorm 1d/2d/4d (#4312)
|
3 years ago |
nihui
|
f770987267
|
pnnx glu batchindex aware conversion (#4285)
|
3 years ago |
nihui
|
777e4ef0a0
|
fix pnnx softmax/normalize/slice negative axis conversion to ncnn (#4284)
|
3 years ago |
Fangjun Kuang
|
5281d51535
|
implement GLU and pnnx conversion (#4283)
|
3 years ago |
nihui
|
77eda4c19f
|
implement lstm proj_size (#4263)
|
3 years ago |
nihui
|
cef95f64a4
|
pnnx math operation (#4251)
|
3 years ago |
miemie2013
|
720f3c9aab
|
Add DeformableConv2D (#4070)
* Add DeformableConv2D
* add unittest and docs
* pnnx torchvision deformconv2d conversion
Co-authored-by: miemie2013 <miemie2013@users.noreply.github.com>
Co-authored-by: nihui <shuizhuyuanluo@126.com>
|
4 years ago |
nihui
|
322667a2ab
|
pnnx fix fused tensor_split operator insert order (#4006)
|
4 years ago |
nihui
|
1892c25360
|
pnnx fuse megvii style shufflechannel slice (#3999)
|
4 years ago |
nihui
|
94786308bd
|
pnnx fuse binaryop eltwise as weighted sum (#4000)
|
4 years ago |
nihui
|
d4a704de0e
|
pnnx eliminate noop upsample (#3991)
|
4 years ago |
nihui
|
b4bae2c9e4
|
pnnx convert torch.tensor_split, fuse full dim size slice to tensor_split (#3988)
|
4 years ago |
nihui
|
044467b2c6
|
pnnx support torch 1.12 (#3981)
|
4 years ago |
nihui
|
f597619e94
|
pnnx export weight inside moduleop (#3902)
|
4 years ago |
nihui
|
5ae827c745
|
convert inplace relu6, match more hardswish pattern (#3952)
* convert inplace relu6, match more hardswish pattern
* upgrade requests
|
4 years ago |
nihui
|
bc8e939c85
|
insert reshape for nn.Linear with 4d/5d inputs (#3959)
* insert reshape for nn.Linear with 4d/5d inputs
* batch index aware reshape, add test
|
4 years ago |
nihui
|
612806a919
|
pnnx convert torch fft family (#3892)
* disable mkl avx/avx2 optimization for potential memory issue in dft
|
4 years ago |
nihui
|
53fa52e78a
|
pnnx support complex data type, convert torch complex/imag/real/fft.rfft2/fft.irfft2 (#3890)
|
4 years ago |
nihui
|
b299729963
|
add another pnnx hardsigmoid pattern (#3881)
|
4 years ago |
nihui
|
7441d8a7c8
|
eliminate noop cat for single input (#3871)
|
4 years ago |
nihui
|
e7a664c6e5
|
convert pnnx torch.index_select and torch.scatter_add (#3842)
|
4 years ago |
nihui
|
026a04f220
|
convert torch.norm to ncnn, fix F.normalize vector (#3828)
|
4 years ago |
nihui
|
2dc1ae45fe
|
pnnx bitwise and compare op (#3791)
* bitwise and compare op
* masked_fill and gather test
|
4 years ago |
nihui
|
0aba8af1d3
|
pnnx swin transformer (#3783)
|
4 years ago |
nihui
|
f9c1787de9
|
implement einsum layer and pnnx conversion (#3768)
|
4 years ago |
nihui
|
ee6402553c
|
layernorm for vector and mat along w, pnnx convnext end2end test (#3764)
|
4 years ago |
nihui
|
318dc4f5f5
|
convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763)
|
4 years ago |
nihui
|
a2b25c88b0
|
pnnx convert torch.einsum (#3744)
* normalize einsum equation
* replace dot pair with alphabet word
* shorten permute equation
|
4 years ago |
nihui
|
57f3017afd
|
pnnx torch bmm max min (#3736)
* convert torch.bmm
* convert torch.min torch.max
|
4 years ago |
nihui
|
a14acbc6dd
|
pnnx save ncnn bin with fp16 storage (#3715)
|
4 years ago |
nihui
|
d42e048b56
|
pnnx convert torch.addmm (#3634)
|
4 years ago |
nihui
|
41cd40ac1f
|
convert pnnx torch.unbind and torch.ones/ones_like family (#3583)
* convert torch.unbind
* torch.ones torch.ones_like
* torch.full torch.full_like
* torch.randn_like
* torch.empty torch.empty_like
|
4 years ago |
nihui
|
2880eff264
|
deconv1d deconv3d (#3584)
* fix sigmoid returns nan with very large input
|
4 years ago |
nihui
|
6b2495cc24
|
add reshape before and after pooling 123d with no batch dimension (#3566)
|
4 years ago |
nihui
|
76e32e9ee6
|
fix interp nearest by scale factor, fix issue #3555 (#3565)
* lets accept the float div vs recp error
|
4 years ago |
nihui
|
d452eca28f
|
convert torch.matmul, eliminate noop pad and identity op, fuse transpose matmul, fuse select to unbind (#3554)
|
4 years ago |
nihui
|
340b4e673e
|
pnnx fold constant (#3521)
|
4 years ago |
nihui
|
c5d7f963b9
|
layer tile (#3491)
|
4 years ago |
nihui
|
7d3503c06a
|
pnnx Tensor index (#3483)
* pnnx Tensor index
* add test
|
4 years ago |
nihuini
|
1db16ce9fc
|
pnnx torch norm stack test
|
4 years ago |