nihui
1172b04355
pnnx pass level2 priority ( #5791 )
* constantlist is constant
* f pad value none
* skip ncnn istft center=False test for old torch
* fix fuse pad conv1d
1 year ago
nihui
0734b657d9
spectrogram and inverse spectrogram ( #5779 )
* only supports hann, hamming and all-one window
* inverse spectrogram does not support length parameter
* spectrogram always returns torch.view_as_real(out) as ncnn does not support complex typed mat yet
* inverse spectrogram always accepts torch.view_as_complex(in) as ncnn does not support complex typed mat yet
1 year ago
nihui
c1f9e959f5
pnnx torch 2.5 ( #5748 )
1 year ago
nihui
80c78a0e40
pnnx fuse t5-layernorm as rmsnorm ( #5675 )
1 year ago
nihui
fdf0df3079
RMSNorm ( #5630 )
1 year ago
nihui
eb6e084c2d
pnnx convert nn.RMSNorm F.rms_norm ( #5628 )
1 year ago
nihui
ecfd88a11b
pnnx2ncnn convert torch.roll with one or two shifts ( #5623 )
1 year ago
nihui
b9debee8fb
pnnx ci for torch 2.4 ( #5618 )
* update onnx proto
1 year ago
nihui
3ee5c18f84
pnnx logaddexp ( #5598 )
1 year ago
nihui
569617f212
pnnx convert onnx expand/permute/repeat/reshape/select/slice/cat/ceil/chunk/flatten/floor/maximum/minimum/split/squeeze/stack/transpose/unbind/unsqueeze ( #5583 )
1 year ago
nihui
e7cae68a22
pnnx convert onnx logsoftmax/logsigmoid/mish/selu/sigmoid/silu/softmin/softplus/softshrink/softsign/tanh/tanhshrink ( #5581 )
1 year ago
nihui
1c40615b2d
pnnx convert onnx sdap reduce min/max/mean/sum/prod ( #5579 )
* pnnx convert onnx sdap
* test reduce
1 year ago
nihui
c59885aeac
pnnx convert onnx multiheadattention ( #5575 )
* pnnx convert onnx multiheadattention
* onnx reducemean reducesum
* reducemax reducemin reduceprod
* mask buggy torch
* avoid shadow output
1 year ago
nihui
854678b5f3
pnnx convert onnx prelu gelu elu leakyrelu relu6 celu hardshrink hardsigmoid hardswish clip ( #5572 )
1 year ago
nihui
d264b6353a
pnnx convert onnx rnn lstm gru ( #5553 )
1 year ago
nihui
74d3eb2345
pnnx convert onnx layernorm instancenorm groupnorm ( #5533 )
* pnnx convert onnx layernorm
* fuse early
* skip layernorm affine false test for torch 2.1
* pnnx convert onnx layernorm instancenorm groupnorm
* take num_features from input shape for instancenorm module
* torch < 1.10 can not handle track_running_stats=True
2 years ago
nihui
98de53920e
pnnx convert onnx lrn, fuse if with constant condition, canonicalize and check foldable inside subgraph ( #5532 )
2 years ago
nihui
755f8e1a6b
pnnx convert onnx batchnorm ( #5529 )
2 years ago
nihui
109ed2665c
pnnx convert onnx avgpool maxpool ( #5527 )
* pnnx convert onnx avgpool
* fuse gather indices
* generalized conv convtranspose
* skip maxpool dilation test for torch < 1.12
* ceil mode for opset 9
2 years ago
nihui
f56b18aaf0
pnnx convert onnx resize upsample ( #5522 )
2 years ago
nihui
2828e7ae96
pnnx reset onnx input shape, convert torch.tile torch.where ( #5517 )
* pnnx reset onnx input shape
* eliminate noop cast
2 years ago
nihui
ffb2fe60ee
pnnx convert onnx pad linear sigmoid softmax relu ( #5516 )
* pnnx convert onnx pad
* pnnx convert onnx linear sigmoid softmax relu
* old onnx softmax
2 years ago
nihui
21babb7eed
pnnx convert onnx conv convtranspose ( #5515 )
2 years ago
nihui
5e9bf90455
pnnx onnx with inputshape ( #5480 )
2 years ago
nihui
f40d5414d5
pnnx build onnx2pnnx ( #5461 )
2 years ago
nihui
84256b1494
pnnx enhance functionize ( #5387 )
* pnnx fix some undefined dtype
* fix ncnn convdw1d dynamic weight loading
2 years ago
nihui
1463e0b07c
pnnx convert torch.slice_scatter ( #5377 )
2 years ago
nihui
a15d491beb
pnnx test torch.arange ( #5376 )
2 years ago
nihui
a2414c7f63
fix pnnx convert torch.stack negative dim ( #5375 )
2 years ago
nihui
40958d3ab3
pnnx support dynamic slice indexes ( #5299 )
* pnnx handle two operands add/sub/rsub variant
* fuse dynamic slice indexes, wip
* pnnx sliceindexes
* reset device may change non-dtype input numeric 5 to 6
* print inf as float
* preserve dtype for generation op
* pnnx convert torch.masked_select
* test masked_select
* test negative slice
2 years ago
nihui
7ed252c854
pnnx handle index_put with empty indices and scalar values ( #5288 )
2 years ago
nihui
6f84952122
pnnx handle more softmin logsoftmax dtype, fuse static full range slices to tensor_split ( #5253 )
* pnnx handle more softmin logsoftmax dtype, fuse static full range slices to tensor_split
* fix convert nn.Conv2d with none bias tensor
* fix embedding input with batch index zero
2 years ago
nihui
09f2723699
pnnx nn.Identity test ( #5238 )
2 years ago
nihui
3785921e01
pnnx convert nn.Upsample with recompute_scale_factor=True ( #5169 )
2 years ago
nihui
4494aadd74
deconvolution dynamic weight ( #5119 )
2 years ago
Yoh
3f437d3f3d
Grid sample op ( #4373 )
* pnnx support grid_sample op
* complete the permute and gridsample operator fusion
* spilt calculation into two stages and support permute fusion
2 years ago
佰阅
bcdc276ffe
add torch.view_as_real and torch.view_as_complex ( #5083 )
2 years ago
Marsyule
69d6051850
add F_pairwise_distance to pnnx and ncnn ( #4942 )
2 years ago
nihui
78aca88d67
elu 4d and selu 4d ( #5047 )
2 years ago
nihui
b8d5a5dbec
pnnx fuse conv3d-bn and deconv3d-bn ( #5045 )
2 years ago
nihui
fdf2c482dc
fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize ( #5043 )
2 years ago
Shu Liqiang
a2103be55d
support ReplicationPad3d to ncnn ( #4919 )
2 years ago
Shu Liqiang
83cca26794
pnnx convert torch.mv ( #4974 )
2 years ago
Beq Jal
bcfec1da33
Celu layer and export to ncnn ( #5019 )
2 years ago
Beq Jal
c851231832
add diag layer and its converter ( #4935 )
2 years ago
Jiang-Weibo
4a2620d1e2
pnnx add torch.cumprod op ( #5002 )
2 years ago
nicochen1118
4923fb955b
pnnx add torch.positive ( #4999 )
2 years ago
莫思潋
379dfd2e29
add torch.lgamma to pnnx ( #4976 )
2 years ago
zyt1024
b3fbbccd8b
pnnx convert torch narrow ( #4918 )
2 years ago
nihui
cb674ac5ed
fix build with toolchain defined _L _U constants ( #4957 )
2 years ago