| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/ |
| H A D | pytorch_qnnpack.h | 46 typedef struct pytorch_qnnp_operator* pytorch_qnnp_operator_t; typedef
70 pytorch_qnnp_operator_t* convolution);
98 pytorch_qnnp_operator_t* convolution);
101 pytorch_qnnp_operator_t convolution,
112 pytorch_qnnp_operator_t convolution,
146 pytorch_qnnp_operator_t* deconvolution);
149 pytorch_qnnp_operator_t deconvolution,
171 pytorch_qnnp_operator_t* fully_connected);
190 pytorch_qnnp_operator_t* fully_connected);
193 pytorch_qnnp_operator_t fully_connected,
[all …]
|
| H A D | qnnpack_func.h | 10 const pytorch_qnnp_operator_t convolution,
119 const pytorch_qnnp_operator_t convolution,
136 const pytorch_qnnp_operator_t deconvolution,
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | indirection.h | 22 pytorch_qnnp_operator_t op,
27 pytorch_qnnp_operator_t op,
31 pytorch_qnnp_operator_t op,
36 pytorch_qnnp_operator_t op,
40 pytorch_qnnp_operator_t op);
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | channel-shuffle.c | 24 pytorch_qnnp_operator_t* channel_shuffle_out) { in pytorch_qnnp_create_channel_shuffle_nc_x8()
25 pytorch_qnnp_operator_t channel_shuffle_op = NULL; in pytorch_qnnp_create_channel_shuffle_nc_x8()
77 pytorch_qnnp_operator_t channel_shuffle_op, in pytorch_qnnp_setup_channel_shuffle_nc_x8()
|
| H A D | clamp.c | 24 pytorch_qnnp_operator_t* clamp_out) { in pytorch_qnnp_create_clamp_nc_u8()
25 pytorch_qnnp_operator_t clamp_op = NULL; in pytorch_qnnp_create_clamp_nc_u8()
78 pytorch_qnnp_operator_t clamp, in pytorch_qnnp_setup_clamp_nc_u8()
|
| H A D | softargmax.c | 25 pytorch_qnnp_operator_t* softargmax_out) { in pytorch_qnnp_create_softargmax_nc_q8()
26 pytorch_qnnp_operator_t softargmax_op = NULL; in pytorch_qnnp_create_softargmax_nc_q8()
115 pytorch_qnnp_operator_t softargmax, in pytorch_qnnp_setup_softargmax_nc_q8()
|
| H A D | add.c | 32 pytorch_qnnp_operator_t* add_out) { in pytorch_qnnp_create_add_nc_q8()
33 pytorch_qnnp_operator_t add_op = NULL; in pytorch_qnnp_create_add_nc_q8()
132 pytorch_qnnp_operator_t add_op, in pytorch_qnnp_setup_add_nc_q8()
|
| H A D | global-average-pooling.c | 30 pytorch_qnnp_operator_t* global_average_pooling_out) { in pytorch_qnnp_create_global_average_pooling_nwc_q8()
31 pytorch_qnnp_operator_t global_average_pooling_op = NULL; in pytorch_qnnp_create_global_average_pooling_nwc_q8()
115 pytorch_qnnp_operator_t global_average_pooling_op, in pytorch_qnnp_setup_global_average_pooling_nwc_q8()
|
| H A D | sigmoid.c | 28 pytorch_qnnp_operator_t* sigmoid_out) { in pytorch_qnnp_create_sigmoid_nc_q8()
29 pytorch_qnnp_operator_t sigmoid_op = NULL; in pytorch_qnnp_create_sigmoid_nc_q8()
135 pytorch_qnnp_operator_t sigmoid, in pytorch_qnnp_setup_sigmoid_nc_q8()
|
| H A D | hardswish.c | 28 pytorch_qnnp_operator_t* hardswish_out) { in pytorch_qnnp_create_hardswish_nc_q8()
29 pytorch_qnnp_operator_t hardswish_op = NULL; in pytorch_qnnp_create_hardswish_nc_q8()
123 pytorch_qnnp_operator_t hardswish, in pytorch_qnnp_setup_hardswish_nc_q8()
|
| H A D | tanh.c | 28 pytorch_qnnp_operator_t* tanh_out) { in pytorch_qnnp_create_tanh_nc_q8()
29 pytorch_qnnp_operator_t tanh_op = NULL; in pytorch_qnnp_create_tanh_nc_q8()
137 pytorch_qnnp_operator_t tanh, in pytorch_qnnp_setup_tanh_nc_q8()
|
| H A D | convolution.c | 69 pytorch_qnnp_operator_t* convolution_out, in pytorch_qnnp_create_convolution_ndhwc_q8()
71 pytorch_qnnp_operator_t convolution = NULL; in pytorch_qnnp_create_convolution_ndhwc_q8()
571 pytorch_qnnp_operator_t* convolution_out) { in pytorch_qnnp_create_convolution2d_nhwc_q8()
628 pytorch_qnnp_operator_t* convolution_out) { in pytorch_qnnp_create_convolution3d_ndhwc_q8()
660 pytorch_qnnp_operator_t convolution, in pytorch_qnnp_setup_convolution2d_nhwc_q8()
683 pytorch_qnnp_operator_t convolution, in pytorch_qnnp_setup_convolution_ndhwc_q8()
|
| H A D | hardsigmoid.c | 28 pytorch_qnnp_operator_t* hardsigmoid_out) { in pytorch_qnnp_create_hardsigmoid_nc_q8()
29 pytorch_qnnp_operator_t hardsigmoid_op = NULL; in pytorch_qnnp_create_hardsigmoid_nc_q8()
140 pytorch_qnnp_operator_t hardsigmoid, in pytorch_qnnp_setup_hardsigmoid_nc_q8()
|
| H A D | leaky-relu.c | 29 pytorch_qnnp_operator_t* leaky_relu_out) { in pytorch_qnnp_create_leaky_relu_nc_q8()
30 pytorch_qnnp_operator_t leaky_relu_op = NULL; in pytorch_qnnp_create_leaky_relu_nc_q8()
145 pytorch_qnnp_operator_t leaky_relu, in pytorch_qnnp_setup_leaky_relu_nc_q8()
|
| H A D | fully-connected.c | 36 pytorch_qnnp_operator_t* fully_connected_out) { in pytorch_qnnp_create_fully_connected_nc_q8()
37 pytorch_qnnp_operator_t fully_connected = NULL; in pytorch_qnnp_create_fully_connected_nc_q8()
131 pytorch_qnnp_operator_t fully_connected, in pytorch_qnnp_setup_fully_connected_nc_q8()
|
| H A D | fully-connected-sparse.c | 41 pytorch_qnnp_operator_t* fully_connected_out) { in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
42 pytorch_qnnp_operator_t fully_connected = NULL; in pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8()
139 pytorch_qnnp_operator_t fully_connected, in pytorch_qnnp_setup_fully_connected_sparse_dq_nc_q8()
|
| H A D | max-pooling.c | 50 pytorch_qnnp_operator_t* max_pooling_out) { in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
51 pytorch_qnnp_operator_t max_pooling = NULL; in pytorch_qnnp_create_max_pooling2d_nhwc_u8()
144 pytorch_qnnp_operator_t max_pooling, in pytorch_qnnp_setup_max_pooling2d_nhwc_u8()
|
| H A D | indirection.c | 18 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_conv3d()
225 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_dwconv()
340 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_deconv2d()
409 pytorch_qnnp_operator_t op, in pytorch_qnnp_indirection_init_maxpool2d()
458 void pytorch_qnnp_indirection_set_step_dimensions(pytorch_qnnp_operator_t op) { in pytorch_qnnp_indirection_set_step_dimensions()
|
| H A D | average-pooling.c | 47 pytorch_qnnp_operator_t* average_pooling_out) { in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
48 pytorch_qnnp_operator_t average_pooling = NULL; in pytorch_qnnp_create_average_pooling2d_nhwc_q8()
205 pytorch_qnnp_operator_t average_pooling, in pytorch_qnnp_setup_average_pooling2d_nhwc_q8()
|
| H A D | deconvolution.c | 61 pytorch_qnnp_operator_t* deconvolution_out) { in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
62 pytorch_qnnp_operator_t deconvolution = NULL; in pytorch_qnnp_create_deconvolution2d_nhwc_q8()
221 pytorch_qnnp_operator_t deconvolution, in pytorch_qnnp_setup_deconvolution2d_nhwc_q8()
|
| H A D | deconv-run.cc | 84 void operator()(pytorch_qnnp_operator_t op) { in operator ()()
90 const pytorch_qnnp_operator_t deconvolution, in qnnpackDeConv()
|
| H A D | operator-delete.c | 15 pytorch_qnnp_operator_t op) { in pytorch_qnnp_delete_operator()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/ |
| H A D | add.cc | 38 pytorch_qnnp_operator_t addOperator = nullptr; in add_nc_q8()
106 pytorch_qnnp_operator_t addOperator = nullptr; in add_nc_q8_inplace()
|
| H A D | global-average-pooling.cc | 45 pytorch_qnnp_operator_t globalPoolingOperator = nullptr; in global_average_pooling_q8()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | QnnpackUtils.h | 21 void operator()(pytorch_qnnp_operator_t op) { in operator()
200 pytorch_qnnp_operator_t convolution{nullptr}; in PackedConvWeightsQnnp()
202 convolution = static_cast<pytorch_qnnp_operator_t>( in PackedConvWeightsQnnp()
|