You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
nihui eceac35a7f
implement MultiheadAttention kdim vdim (#4347)
3 years ago
..
CMakeLists.txt gridsample op support (#4288) 3 years ago
test_F_adaptive_avg_pool1d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_adaptive_avg_pool2d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_adaptive_avg_pool3d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_adaptive_max_pool1d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_adaptive_max_pool2d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_adaptive_max_pool3d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_alpha_dropout.py convert pnnx clone normal 4 years ago
test_F_avg_pool1d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_F_avg_pool2d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_F_avg_pool3d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_F_batch_norm.py pnnx user provided parameter (#3417) 4 years ago
test_F_conv1d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_F_conv2d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_F_conv3d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_F_conv_transpose1d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_F_conv_transpose2d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_F_conv_transpose3d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_F_dropout.py convert pnnx clone normal 4 years ago
test_F_dropout2d.py convert pnnx clone normal 4 years ago
test_F_dropout3d.py convert pnnx clone normal 4 years ago
test_F_elu.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_embedding.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_F_feature_alpha_dropout.py convert pnnx clone normal 4 years ago
test_F_fold.py implement ncnn fold and unfold (#4326) 3 years ago
test_F_gelu.py dynamic convolution weight (#3408) 4 years ago
test_F_glu.py pnnx glu batchindex aware conversion (#4285) 3 years ago
test_F_grid_sample.py gridsample op support (#4288) 3 years ago
test_F_group_norm.py groupnorm 1d/2d/4d (#4312) 3 years ago
test_F_hardsigmoid.py dynamic convolution weight (#3408) 4 years ago
test_F_hardswish.py dynamic convolution weight (#3408) 4 years ago
test_F_hardtanh.py dynamic convolution weight (#3408) 4 years ago
test_F_interpolate.py fix interp nearest by scale factor, fix issue #3555 (#3565) 4 years ago
test_F_layer_norm.py layernorm for vector and mat along w, pnnx convnext end2end test (#3764) 4 years ago
test_F_leaky_relu.py dynamic convolution weight (#3408) 4 years ago
test_F_local_response_norm.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_max_pool1d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_F_max_pool2d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_F_max_pool3d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_F_mish.py dynamic convolution weight (#3408) 4 years ago
test_F_normalize.py convert torch.norm to ncnn, fix F.normalize vector (#3828) 4 years ago
test_F_pad.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_pixel_shuffle.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_pixel_unshuffle.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_prelu.py pnnx export weight inside moduleop (#3902) 3 years ago
test_F_relu.py dynamic convolution weight (#3408) 4 years ago
test_F_relu6.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_sigmoid.py dynamic convolution weight (#3408) 4 years ago
test_F_silu.py dynamic convolution weight (#3408) 4 years ago
test_F_softmax.py fix pnnx softmax/normalize/slice negative axis conversion to ncnn (#4284) 3 years ago
test_F_tanh.py dynamic convolution weight (#3408) 4 years ago
test_F_unfold.py implement ncnn fold and unfold (#4326) 3 years ago
test_F_upsample.py fix interp nearest by scale factor, fix issue #3555 (#3565) 4 years ago
test_F_upsample_bilinear.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_upsample_nearest.py fix interp nearest by scale factor, fix issue #3555 (#3565) 4 years ago
test_Tensor_contiguous.py convert pnnx clone normal 4 years ago
test_Tensor_repeat.py layer tile (#3491) 4 years ago
test_Tensor_reshape.py convert pnnx argmax argmin, fix #3435 4 years ago
test_Tensor_slice.py dynamic convolution weight (#3408) 4 years ago
test_Tensor_view.py convert pnnx argmax argmin, fix #3435 4 years ago
test_convnext_tiny.py layernorm for vector and mat along w, pnnx convnext end2end test (#3764) 4 years ago
test_mobilenet_v2.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_mobilenet_v3_small.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_ncnn_fuse_binaryop_eltwise.py pnnx fuse binaryop eltwise as weighted sum (#4000) 3 years ago
test_ncnn_fuse_shufflechannel_slice.py pnnx fuse megvii style shufflechannel slice (#3999) 3 years ago
test_ncnn_fuse_transpose_matmul.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_AdaptiveAvgPool1d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_AdaptiveAvgPool2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_AdaptiveAvgPool3d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_AdaptiveMaxPool1d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_AdaptiveMaxPool2d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_AdaptiveMaxPool3d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_AlphaDropout.py convert pnnx clone normal 4 years ago
test_nn_AvgPool1d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_AvgPool2d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_nn_AvgPool3d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_nn_BatchNorm1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_BatchNorm2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_BatchNorm3d.py relu3d, batchnorm3d, reshape4d, flatten4d, permute4d (#3397) 4 years ago
test_nn_ChannelShuffle.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ConstantPad1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ConstantPad2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ConstantPad3d.py initial data structure changes for 3dcnn, conv3d, pooling3d (#3378) 4 years ago
test_nn_Conv1d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_nn_Conv2d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_nn_Conv3d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_nn_ConvTranspose1d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_ConvTranspose2d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_ConvTranspose3d.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_Dropout.py convert pnnx clone normal 4 years ago
test_nn_Dropout2d.py convert pnnx clone normal 4 years ago
test_nn_Dropout3d.py convert pnnx clone normal 4 years ago
test_nn_ELU.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_Embedding.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_Fold.py implement ncnn fold and unfold (#4326) 3 years ago
test_nn_GELU.py dynamic convolution weight (#3408) 4 years ago
test_nn_GLU.py pnnx glu batchindex aware conversion (#4285) 3 years ago
test_nn_GRU.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_GroupNorm.py groupnorm 1d/2d/4d (#4312) 3 years ago
test_nn_Hardsigmoid.py dynamic convolution weight (#3408) 4 years ago
test_nn_Hardswish.py dynamic convolution weight (#3408) 4 years ago
test_nn_Hardtanh.py dynamic convolution weight (#3408) 4 years ago
test_nn_InstanceNorm2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_LSTM.py implement lstm proj_size (#4263) 3 years ago
test_nn_LayerNorm.py layernorm for vector and mat along w, pnnx convnext end2end test (#3764) 4 years ago
test_nn_LeakyReLU.py dynamic convolution weight (#3408) 4 years ago
test_nn_Linear.py insert reshape for nn.Linear with 4d/5d inputs (#3959) 3 years ago
test_nn_LocalResponseNorm.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_MaxPool1d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_nn_MaxPool2d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_nn_MaxPool3d.py add reshape before and after pooling 123d with no batch dimension (#3566) 4 years ago
test_nn_Mish.py dynamic convolution weight (#3408) 4 years ago
test_nn_MultiheadAttention.py implement MultiheadAttention kdim vdim (#4347) 3 years ago
test_nn_PReLU.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_PixelShuffle.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_PixelUnshuffle.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_RNN.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_nn_ReLU.py dynamic convolution weight (#3408) 4 years ago
test_nn_ReLU6.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ReflectionPad1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ReflectionPad2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ReplicationPad1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_ReplicationPad2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_SELU.py dynamic convolution weight (#3408) 4 years ago
test_nn_SiLU.py dynamic convolution weight (#3408) 4 years ago
test_nn_Sigmoid.py dynamic convolution weight (#3408) 4 years ago
test_nn_Softmax.py fix pnnx softmax/normalize/slice negative axis conversion to ncnn (#4284) 3 years ago
test_nn_Softmax2d.py pnnx convert nn.Softmax2d (#4324) 3 years ago
test_nn_Tanh.py dynamic convolution weight (#3408) 4 years ago
test_nn_Unfold.py implement ncnn fold and unfold (#4326) 3 years ago
test_nn_Upsample.py fix interp nearest by scale factor, fix issue #3555 (#3565) 4 years ago
test_nn_UpsamplingBilinear2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_UpsamplingNearest2d.py fix interp nearest by scale factor, fix issue #3555 (#3565) 4 years ago
test_nn_ZeroPad2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_resnet18.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_shufflenet_v2_x1_0.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_squeezenet1_1.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_torch_abs.py pnnx math operation (#4251) 3 years ago
test_torch_acos.py pnnx math operation (#4251) 3 years ago
test_torch_addmm.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_torch_amax.py reduction4d, merge keepdims arg, add test (#3469) 4 years ago
test_torch_amin.py reduction4d, merge keepdims arg, add test (#3469) 4 years ago
test_torch_asin.py pnnx math operation (#4251) 3 years ago
test_torch_atan.py pnnx math operation (#4251) 3 years ago
test_torch_bmm.py pnnx torch bmm max min (#3736) 4 years ago
test_torch_cat.py convert pnnx clone normal 4 years ago
test_torch_ceil.py pnnx math operation (#4251) 3 years ago
test_torch_chunk.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_torch_clamp.py convert pnnx clone normal 4 years ago
test_torch_clone.py convert pnnx clone normal 4 years ago
test_torch_cos.py pnnx math operation (#4251) 3 years ago
test_torch_einsum.py implement einsum layer and pnnx conversion (#3768) 4 years ago
test_torch_exp.py pnnx math operation (#4251) 3 years ago
test_torch_floor.py pnnx math operation (#4251) 3 years ago
test_torch_log.py pnnx math operation (#4251) 3 years ago
test_torch_logsumexp.py reduction4d, merge keepdims arg, add test (#3469) 4 years ago
test_torch_matmul.py pnnx save ncnn bin with fp16 storage (#3715) 4 years ago
test_torch_max.py pnnx torch bmm max min (#3736) 4 years ago
test_torch_mean.py reduction4d, merge keepdims arg, add test (#3469) 4 years ago
test_torch_min.py pnnx torch bmm max min (#3736) 4 years ago
test_torch_neg.py pnnx math operation (#4251) 3 years ago
test_torch_norm.py convert torch.norm to ncnn, fix F.normalize vector (#3828) 4 years ago
test_torch_permute.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_torch_pow.py pnnx math operation (#4251) 3 years ago
test_torch_prod.py reduction4d, merge keepdims arg, add test (#3469) 4 years ago
test_torch_reciprocal.py pnnx math operation (#4251) 3 years ago
test_torch_rsqrt.py pnnx math operation (#4251) 3 years ago
test_torch_sin.py pnnx math operation (#4251) 3 years ago
test_torch_sqrt.py pnnx math operation (#4251) 3 years ago
test_torch_square.py pnnx math operation (#4251) 3 years ago
test_torch_squeeze.py squeeze and expanddims 4d (#4346) 3 years ago
test_torch_sum.py reduction4d, merge keepdims arg, add test (#3469) 4 years ago
test_torch_tan.py pnnx math operation (#4251) 3 years ago
test_torch_tanh.py pnnx math operation (#4251) 3 years ago
test_torch_tensor_split.py pnnx convert torch.tensor_split, fuse full dim size slice to tensor_split (#3988) 3 years ago
test_torch_transpose.py convert pnnx argmax argmin, fix #3435 4 years ago
test_torch_unbind.py convert pnnx torch.unbind and torch.ones/ones_like family (#3583) 4 years ago
test_torch_unsqueeze.py squeeze and expanddims 4d (#4346) 3 years ago
test_torchvision_DeformConv2d.py Add DeformableConv2D (#4070) 3 years ago
test_vit_b_32.py pnnx support torch 1.12 (#3981) 3 years ago