65 Commits (057b5bb515d551fa64decdb7350422c19feba447)

Author SHA1 Message Date
  nihui aed05aa851
pnnx fuse more function to module (#4351) 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) 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) 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) 4 years ago
  nihui bc8e939c85
insert reshape for nn.Linear with 4d/5d inputs (#3959) 4 years ago
  nihui 612806a919
pnnx convert torch fft family (#3892) 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) 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) 4 years ago
  nihui 57f3017afd
pnnx torch bmm max min (#3736) 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) 4 years ago
  nihui 2880eff264
deconv1d deconv3d (#3584) 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) 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) 4 years ago
  nihuini 1db16ce9fc
pnnx torch norm stack test 4 years ago