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 8363040cb4
pnnx ncnn gelu fast mode, fix interp 2d resize (#5999)
1 year ago
..
ncnn pnnx ncnn gelu fast mode, fix interp 2d resize (#5999) 1 year ago
onnx pnnx convert onnx reducel1 reducel2, add onnx norm test (#5988) 1 year ago
CMakeLists.txt spectrogram and inverse spectrogram (#5779) 1 year ago
run_test.cmake PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_adaptive_avg_pool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_adaptive_avg_pool2d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_F_adaptive_avg_pool3d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_F_adaptive_max_pool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_adaptive_max_pool2d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_F_adaptive_max_pool3d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_F_affine_grid.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_alpha_dropout.py convert dropout family (#3425) 4 years ago
test_F_avg_pool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 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_celu.py elu 4d and selu 4d (#5047) 2 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 convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_F_conv_transpose1d.py pnnx user provided parameter (#3417) 4 years ago
test_F_conv_transpose2d.py pnnx user provided parameter (#3417) 4 years ago
test_F_conv_transpose3d.py deconv1d deconv3d (#3584) 4 years ago
test_F_dropout.py convert dropout family (#3425) 4 years ago
test_F_dropout2d.py convert dropout family (#3425) 4 years ago
test_F_dropout3d.py convert dropout family (#3425) 4 years ago
test_F_elu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_embedding.py pnnx cmdline argument inputshape with type (#3419) 4 years ago
test_F_feature_alpha_dropout.py convert dropout family (#3425) 4 years ago
test_F_fold.py implement ncnn fold and unfold (#4326) 3 years ago
test_F_gelu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_glu.py implement GLU and pnnx conversion (#4283) 3 years ago
test_F_grid_sample.py gridsample op support (#4288) 3 years ago
test_F_group_norm.py pnnx user provided parameter (#3417) 4 years ago
test_F_hardshrink.py elu 4d and selu 4d (#5047) 2 years ago
test_F_hardsigmoid.py elu 4d and selu 4d (#5047) 2 years ago
test_F_hardswish.py elu 4d and selu 4d (#5047) 2 years ago
test_F_hardtanh.py elu 4d and selu 4d (#5047) 2 years ago
test_F_instance_norm.py pnnx user provided parameter (#3417) 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 pnnx user provided parameter (#3417) 4 years ago
test_F_leaky_relu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_linear.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_local_response_norm.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_log_softmax.py elu 4d and selu 4d (#5047) 2 years ago
test_F_logsigmoid.py elu 4d and selu 4d (#5047) 2 years ago
test_F_lp_pool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_lp_pool2d.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 pnnx build onnx2pnnx (#5461) 2 years ago
test_F_normalize.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_pad.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_F_pairwise_distance.py add F_pairwise_distance to pnnx and ncnn (#4942) 2 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 elu 4d and selu 4d (#5047) 2 years ago
test_F_relu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_relu6.py elu 4d and selu 4d (#5047) 2 years ago
test_F_rms_norm.py pnnx convert nn.RMSNorm F.rms_norm (#5628) 1 year ago
test_F_rrelu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_scaled_dot_product_attention.py pnnx build onnx2pnnx (#5461) 2 years ago
test_F_selu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_sigmoid.py elu 4d and selu 4d (#5047) 2 years ago
test_F_silu.py elu 4d and selu 4d (#5047) 2 years ago
test_F_softmax.py elu 4d and selu 4d (#5047) 2 years ago
test_F_softmin.py elu 4d and selu 4d (#5047) 2 years ago
test_F_softplus.py elu 4d and selu 4d (#5047) 2 years ago
test_F_softshrink.py elu 4d and selu 4d (#5047) 2 years ago
test_F_softsign.py elu 4d and selu 4d (#5047) 2 years ago
test_F_tanh.py elu 4d and selu 4d (#5047) 2 years ago
test_F_tanhshrink.py elu 4d and selu 4d (#5047) 2 years ago
test_F_threshold.py elu 4d and selu 4d (#5047) 2 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 PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_Tensor_expand.py multiheadattention attn mask (#4668) 3 years ago
test_Tensor_fill.py pnnx update (#4870) 2 years ago
test_Tensor_index.py pnnx fold constant (#3521) 4 years ago
test_Tensor_index_put.py pnnx handle index_put with empty indices and scalar values (#5288) 2 years ago
test_Tensor_masked_fill.py pnnx bitwise and compare op (#3791) 4 years ago
test_Tensor_new_empty.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_Tensor_new_full.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_Tensor_new_ones.py fold new_full and full_like (#4323) 3 years ago
test_Tensor_new_zeros.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_Tensor_permute.py pnnx build onnx2pnnx (#5461) 2 years ago
test_Tensor_repeat.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_Tensor_reshape.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_Tensor_select.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_Tensor_slice.py crop4d (#3402) 4 years ago
test_Tensor_slice_copy.py pnnx enhance functionize (#5387) 2 years ago
test_Tensor_to.py pnnx add missing Tensor.to pattern (#4908) 2 years ago
test_Tensor_type_as.py pnnx update (#4870) 2 years ago
test_Tensor_view.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_convnext_tiny.py pnnx auto inputshape from traced inputs (#5825) 1 year ago
test_ir_complex.py pnnx build onnx2pnnx (#5461) 2 years ago
test_mobilenet_v2.py pnnx auto inputshape from traced inputs (#5825) 1 year ago
test_mobilenet_v3_small.py pnnx auto inputshape from traced inputs (#5825) 1 year ago
test_nn_AdaptiveAvgPool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_AdaptiveAvgPool2d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_nn_AdaptiveAvgPool3d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_nn_AdaptiveMaxPool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_AdaptiveMaxPool2d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_nn_AdaptiveMaxPool3d.py fuse adaptive pool dynamic output size, implement ncnn adaptive pooling dynamic outsize (#5043) 2 years ago
test_nn_AlphaDropout.py convert dropout family (#3425) 4 years ago
test_nn_AvgPool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 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 PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_CELU.py elu 4d and selu 4d (#5047) 2 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 PNNX is an open standard for PyTorch model interoperability (#3262) 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 pnnx convert conv with non-zero padding mode (#4849) 2 years ago
test_nn_ConvTranspose1d.py pnnx reset maxpool return_indices if only output found, drop convtransposed output_size arg for static output padding (#4654) 3 years ago
test_nn_ConvTranspose2d.py pnnx reset maxpool return_indices if only output found, drop convtransposed output_size arg for static output padding (#4654) 3 years ago
test_nn_ConvTranspose3d.py pnnx reset maxpool return_indices if only output found, drop convtransposed output_size arg for static output padding (#4654) 3 years ago
test_nn_Dropout.py convert dropout family (#3425) 4 years ago
test_nn_Dropout2d.py convert dropout family (#3425) 4 years ago
test_nn_Dropout3d.py convert dropout family (#3425) 4 years ago
test_nn_ELU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Embedding.py pnnx cmdline argument inputshape with type (#3419) 4 years ago
test_nn_Fold.py implement ncnn fold and unfold (#4326) 3 years ago
test_nn_GELU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_GLU.py implement GLU and pnnx conversion (#4283) 3 years ago
test_nn_GRU.py fix pnnx gru rnn with optional output, fix #4608 (#4631) 3 years ago
test_nn_GroupNorm.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_Hardshrink.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Hardsigmoid.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Hardswish.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Hardtanh.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Identity.py pnnx nn.Identity test (#5238) 2 years ago
test_nn_InstanceNorm1d.py pnnx convert onnx layernorm instancenorm groupnorm (#5533) 1 year ago
test_nn_InstanceNorm2d.py pnnx convert onnx layernorm instancenorm groupnorm (#5533) 1 year ago
test_nn_InstanceNorm3d.py pnnx convert onnx layernorm instancenorm groupnorm (#5533) 1 year ago
test_nn_LPPool1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_LPPool2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_LSTM.py fix pnnx gru rnn with optional output, fix #4608 (#4631) 3 years ago
test_nn_LayerNorm.py pnnx convert onnx layernorm instancenorm groupnorm (#5533) 1 year ago
test_nn_LeakyReLU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Linear.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_LocalResponseNorm.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_LogSigmoid.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_LogSoftmax.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_MaxPool1d.py pnnx reset maxpool return_indices if only output found, drop convtransposed output_size arg for static output padding (#4654) 3 years ago
test_nn_MaxPool2d.py pnnx reset maxpool return_indices if only output found, drop convtransposed output_size arg for static output padding (#4654) 3 years ago
test_nn_MaxPool3d.py pnnx reset maxpool return_indices if only output found, drop convtransposed output_size arg for static output padding (#4654) 3 years ago
test_nn_Mish.py pnnx build onnx2pnnx (#5461) 2 years ago
test_nn_MultiheadAttention.py pnnx fuse chinese clip mha and write need_weights=False (#4745) 3 years ago
test_nn_PReLU.py elu 4d and selu 4d (#5047) 2 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_RMSNorm.py pnnx convert nn.RMSNorm F.rms_norm (#5628) 1 year ago
test_nn_RNN.py fix pnnx gru rnn with optional output, fix #4608 (#4631) 3 years ago
test_nn_RReLU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_ReLU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_ReLU6.py elu 4d and selu 4d (#5047) 2 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_ReplicationPad3d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_nn_SELU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_SiLU.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Sigmoid.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Softmax.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Softmax2d.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Softmin.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Softplus.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Softshrink.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Softsign.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Tanh.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Tanhshrink.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Threshold.py elu 4d and selu 4d (#5047) 2 years ago
test_nn_Unfold.py implement ncnn fold and unfold (#4326) 3 years ago
test_nn_Upsample.py pnnx convert nn.Upsample with recompute_scale_factor=True (#5169) 2 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_pnnx_eliminate_noop_cat.py eliminate noop cat for single input (#3871) 4 years ago
test_pnnx_eliminate_noop_expand.py pnnx eliminate noop expand and expand_as (#4850) 2 years ago
test_pnnx_eliminate_noop_math.py pnnx fold constant (#3521) 4 years ago
test_pnnx_eliminate_noop_upsample.py pnnx eliminate noop upsample (#3991) 3 years ago
test_pnnx_expression.py pnnx fuse expression for scalar-like attribute and unbind chain (#4928) 2 years ago
test_pnnx_fold_constant.py pnnx fold constant (#3521) 4 years ago
test_pnnx_fuse_adjacent_reshape.py pnnx fuse more function to module (#4351) 3 years ago
test_pnnx_fuse_conv1d_batchnorm1d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_pnnx_fuse_conv2d_batchnorm2d.py convert Tensor.select to ncnn, fuse multiheadattention unpack, add vit_b_32 end2end test (#3763) 4 years ago
test_pnnx_fuse_conv3d_batchnorm3d.py pnnx fuse conv3d-bn and deconv3d-bn (#5045) 2 years ago
test_pnnx_fuse_convtranspose1d_batchnorm1d.py pnnx fuse conv1d-bn convtranspose1d-bn 4 years ago
test_pnnx_fuse_convtranspose2d_batchnorm2d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_pnnx_fuse_convtranspose3d_batchnorm3d.py pnnx fuse conv3d-bn and deconv3d-bn (#5045) 2 years ago
test_pnnx_fuse_input_unpack.py pnnx flatten input tuple list (#4498) 3 years ago
test_pnnx_fuse_layernorm.py pnnx rewrite multiple ops (#4780) 3 years ago
test_pnnx_fuse_linear_batchnorm1d.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_pnnx_fuse_multiheadattention.py pnnx build onnx2pnnx (#5461) 2 years ago
test_pnnx_fuse_pad_conv1d.py pnnx fuse more function to module (#4351) 3 years ago
test_pnnx_fuse_pad_conv2d.py pnnx fuse more function to module (#4351) 3 years ago
test_pnnx_fuse_pixel_unshuffle.py Fuse pixel unshuffle (#4504) 3 years ago
test_pnnx_fuse_rmsnorm.py pnnx fuse t5-layernorm as rmsnorm (#5675) 1 year ago
test_pnnx_fuse_scaled_dot_product_attention.py pnnx build onnx2pnnx (#5461) 2 years ago
test_pnnx_fuse_select_to_unbind.py pnnx fix fused tensor_split operator insert order (#4006) 3 years ago
test_pnnx_fuse_slice_to_tensor_split.py pnnx handle more softmin logsoftmax dtype, fuse static full range slices to tensor_split (#5253) 2 years ago
test_quantization_shufflenet_v2_x1_0.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_resnet18.py pnnx auto inputshape from traced inputs (#5825) 1 year ago
test_shufflenet_v2_x1_0.py pnnx auto inputshape from traced inputs (#5825) 1 year ago
test_squeezenet1_1.py pnnx auto inputshape from traced inputs (#5825) 1 year ago
test_swin_t.py pnnx auto inputshape from traced inputs (#5825) 1 year 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_acosh.py pnnx math operation (#4251) 3 years ago
test_torch_addmm.py pnnx convert torch.addmm (#3634) 4 years ago
test_torch_amax.py convert some pnnx reduction family 4 years ago
test_torch_amin.py convert some pnnx reduction family 4 years ago
test_torch_arange.py pnnx test torch.arange (#5376) 2 years ago
test_torch_argmax.py convert pnnx argmax argmin, fix #3435 4 years ago
test_torch_argmin.py convert pnnx argmax argmin, fix #3435 4 years ago
test_torch_asin.py pnnx math operation (#4251) 3 years ago
test_torch_asinh.py pnnx math operation (#4251) 3 years ago
test_torch_atan.py pnnx math operation (#4251) 3 years ago
test_torch_atan2.py pnnx math operation (#4251) 3 years ago
test_torch_atanh.py pnnx math operation (#4251) 3 years ago
test_torch_bitwise_and.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_bitwise_left_shift.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_bitwise_not.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_bitwise_or.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_bitwise_right_shift.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_bitwise_xor.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_bmm.py pnnx torch bmm max min (#3736) 4 years ago
test_torch_cat.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_ceil.py pnnx math operation (#4251) 3 years ago
test_torch_chunk.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_clamp.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_clone.py convert pnnx clone normal 4 years ago
test_torch_complex.py pnnx support complex data type, convert torch complex/imag/real/fft.rfft2/fft.irfft2 (#3890) 4 years ago
test_torch_cos.py pnnx math operation (#4251) 3 years ago
test_torch_cosh.py pnnx math operation (#4251) 3 years ago
test_torch_cross.py pnnx convert torch cross and t (#4896) 2 years ago
test_torch_cumprod.py pnnx add torch.cumprod op (#5002) 2 years ago
test_torch_cumsum.py Support torch.cumsum (#4505) 3 years ago
test_torch_diag.py add diag layer and its converter (#4935) 2 years ago
test_torch_einsum.py pnnx update (#4870) 2 years ago
test_torch_eq.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_exp.py pnnx math operation (#4251) 3 years ago
test_torch_fft_fft.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_fft2.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_fftn.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_hfft.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_hfft2.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_fft_hfftn.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_fft_ifft.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_ifft2.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_ifftn.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_ihfft.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_ihfft2.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_fft_ihfftn.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_fft_irfft.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_irfft2.py pnnx support complex data type, convert torch complex/imag/real/fft.rfft2/fft.irfft2 (#3890) 4 years ago
test_torch_fft_irfftn.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_rfft.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_fft_rfft2.py pnnx support complex data type, convert torch complex/imag/real/fft.rfft2/fft.irfft2 (#3890) 4 years ago
test_torch_fft_rfftn.py pnnx convert torch fft family (#3892) 4 years ago
test_torch_flatten.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_floor.py pnnx math operation (#4251) 3 years ago
test_torch_full.py convert pnnx torch.unbind and torch.ones/ones_like family (#3583) 4 years ago
test_torch_full_like.py convert pnnx torch.unbind and torch.ones/ones_like family (#3583) 4 years ago
test_torch_gather.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_ge.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_gt.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_imag.py pnnx support complex data type, convert torch complex/imag/real/fft.rfft2/fft.irfft2 (#3890) 4 years ago
test_torch_index_select.py convert pnnx torch.index_select and torch.scatter_add (#3842) 4 years ago
test_torch_istft.py spectrogram and inverse spectrogram (#5779) 1 year ago
test_torch_le.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_lgamma.py add torch.lgamma to pnnx (#4976) 2 years ago
test_torch_log.py pnnx math operation (#4251) 3 years ago
test_torch_log10.py pnnx convert torch log10, pow 2 as square (#4518) 3 years ago
test_torch_logaddexp.py pnnx logaddexp (#5598) 1 year ago
test_torch_logsumexp.py convert some pnnx reduction family 4 years ago
test_torch_lt.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_masked_select.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_torch_matmul.py convert torch.matmul, eliminate noop pad and identity op, fuse transpose matmul, fuse select to unbind (#3554) 4 years ago
test_torch_max.py fix build with toolchain defined _L _U constants (#4957) 2 years ago
test_torch_maximum.py pnnx convert torch maximum minimum and torch max min as expression (#4944) 2 years ago
test_torch_mean.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_min.py fix build with toolchain defined _L _U constants (#4957) 2 years ago
test_torch_minimum.py pnnx convert torch maximum minimum and torch max min as expression (#4944) 2 years ago
test_torch_mm.py pnnx convert torch.mm (#4589) 3 years ago
test_torch_mv.py pnnx convert torch.mv (#4974) 2 years ago
test_torch_narrow.py pnnx convert torch narrow (#4918) 2 years ago
test_torch_ne.py pnnx bitwise and compare op (#3791) 4 years ago
test_torch_neg.py pnnx math operation (#4251) 3 years ago
test_torch_norm.py pnnx torch norm stack test 4 years ago
test_torch_ones.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_torch_ones_like.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_torch_positive.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_pow.py pnnx math operation (#4251) 3 years ago
test_torch_prod.py convert some pnnx reduction family 4 years ago
test_torch_real.py pnnx support complex data type, convert torch complex/imag/real/fft.rfft2/fft.irfft2 (#3890) 4 years ago
test_torch_reciprocal.py pnnx math operation (#4251) 3 years ago
test_torch_repeat_interleave.py pnnx rewrite multiple ops (#4780) 3 years ago
test_torch_roll.py pnnx2ncnn convert torch.roll with one or two shifts (#5623) 1 year ago
test_torch_round.py pnnx convert torch round trunc (#4813) 2 years ago
test_torch_rsqrt.py pnnx math operation (#4251) 3 years ago
test_torch_scatter_add.py convert pnnx torch.index_select and torch.scatter_add (#3842) 4 years ago
test_torch_sign.py pnnx math operation (#4251) 3 years ago
test_torch_sin.py pnnx math operation (#4251) 3 years ago
test_torch_sinh.py pnnx math operation (#4251) 3 years ago
test_torch_slice_scatter.py pnnx build onnx2pnnx (#5461) 2 years ago
test_torch_split.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 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 PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_stack.py fix pnnx convert torch.stack negative dim (#5375) 2 years ago
test_torch_std.py pnnx convert torch std (#4715) 3 years ago
test_torch_stft.py spectrogram and inverse spectrogram (#5779) 1 year ago
test_torch_sum.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_t.py pnnx convert torch cross and t (#4896) 2 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 slice 4d (#4525) 3 years ago
test_torch_tile.py pnnx reset onnx input shape, convert torch.tile torch.where (#5517) 1 year ago
test_torch_topk.py pnnx update (#4870) 2 years ago
test_torch_transpose.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_trunc.py pnnx math operation (#4251) 3 years ago
test_torch_unbind.py 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
test_torch_unsqueeze.py PNNX is an open standard for PyTorch model interoperability (#3262) 4 years ago
test_torch_view_as_complex.py add torch.view_as_real and torch.view_as_complex (#5083) 2 years ago
test_torch_view_as_real.py add torch.view_as_real and torch.view_as_complex (#5083) 2 years ago
test_torch_where.py pnnx reset onnx input shape, convert torch.tile torch.where (#5517) 1 year ago
test_torch_zeros.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_torch_zeros_like.py pnnx support dynamic slice indexes (#5299) 2 years ago
test_torchaudio_F_inverse_spectrogram.py spectrogram and inverse spectrogram (#5779) 1 year ago
test_torchaudio_F_spectrogram.py pnnx pass level2 priority (#5791) 1 year ago
test_torchaudio_InverseSpectrogram.py spectrogram and inverse spectrogram (#5779) 1 year ago
test_torchaudio_Spectrogram.py pnnx pass level2 priority (#5791) 1 year ago
test_torchvision_DeformConv2d.py Add DeformableConv2D (#4070) 3 years ago
test_torchvision_RoIAlign.py pnnx roialign (#3478) 4 years ago
test_vit_b_32.py pnnx auto inputshape from traced inputs (#5825) 1 year ago