|
pytorch
1.8.2
About: PyTorch provides Tensor computation (like NumPy) with strong GPU acceleration and Deep Neural Networks (in Python) built on a tape-based autograd system. LTS (Long Term Support) release.
 Fossies Dox: pytorch-1.8.2.tar.gz ("unofficial" and yet experimental doxygen-generated source code documentation)  |
#include <stdbool.h>#include <stddef.h>#include <stdint.h>#include <qnnpack/common.h>#include <cpuinfo.h>Go to the source code of this file.
Typedefs | |
| typedef void(* | pytorch_q8gemm_ukernel_function) (size_t mr, size_t nr, size_t k, const uint8_t *a, size_t a_stride, const void *w, uint8_t *c, size_t c_stride, size_t output_channel_index, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gemm_dq_ukernel_function) (size_t mr, size_t nr, size_t k, const uint8_t *a, size_t a_stride, const void *w, const float *bias, float *c, size_t c_stride, size_t output_channel_index, const struct pytorch_qnnp_conv_dynamic_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gemm_dq_sparse_ukernel_function) (size_t mr, size_t nr, const uint8_t *a, size_t a_stride, const uint8_t *packed_w, const uint32_t *w_row_ptr, const uint32_t *w_block_ids_ptr, const float *bias, float *c, size_t c_stride, size_t output_channel_index, const struct pytorch_qnnp_conv_dynamic_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gemm_dq_sparse_packedA_ukernel_function) (size_t mr, size_t nr, const uint8_t *a_packed, const uint8_t *packed_w, const uint32_t *w_row_ptr, const uint32_t *w_block_ids_ptr, const float *bias, float *c, size_t c_stride, size_t output_channel_index, const struct pytorch_qnnp_conv_dynamic_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gemm_sparse_packA_ukernel_function) (const size_t mr, const size_t K, const uint8_t *a, const size_t a_stride, uint8_t *a_packed) |
| typedef void(* | pytorch_q8conv_ukernel_function) (size_t mr, size_t nr, size_t kc, size_t ks, const uint8_t **a, const void *w, uint8_t *c, size_t c_stride, size_t output_channel_index, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gemm_xzp_ukernel_function) (size_t mr, size_t nr, size_t k, const uint8_t *a, size_t a_stride, const int32_t *a_sum, const void *w, uint8_t *c, size_t c_stride, const union pytorch_qnnp_q31_requantization_params *requantization_params) |
| typedef void(* | pytorch_q8sum_rows_ukernel_function) (const uint8_t *a, size_t m, size_t k, size_t stride, int32_t multiplier, int32_t *sums) |
| typedef void(* | pytorch_xzipc_ukernel_function) (size_t n, const void *x, void *y) |
| typedef void(* | pytorch_xzipv_ukernel_function) (size_t n, size_t m, const void *x, void *y) |
| typedef void(* | pytorch_x8lut_ukernel_function) (size_t n, const uint8_t *x, const uint8_t *t, uint8_t *y) |
| typedef void(* | pytorch_sgemm_ukernel_function) (size_t mr, size_t nr, size_t k, const float *a, size_t a_stride, const float *w, float *c, size_t c_stride, const struct pytorch_qnnp_fp32_clamping_params *clamping_params) |
| typedef void(* | pytorch_sconv_ukernel_function) (size_t mr, size_t nr, size_t kc, size_t ks, const float **a, const float *w, float *c, size_t c_stride, const struct pytorch_qnnp_fp32_clamping_params *clamping_params) |
| typedef void(* | pytorch_hgemm_ukernel_function) (size_t mr, size_t nr, size_t k, const void *a, size_t a_stride, const void *w, void *c, size_t c_stride, const struct pytorch_qnnp_fp16_clamping_params *clamping_params) |
| typedef void(* | pytorch_q8dwconv_up_ukernel_function) (size_t channels, size_t output_width, const uint8_t **input, const void *weights, uint8_t *output, size_t input_stride, size_t output_increment, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8dwconv_mp_ukernel_function) (size_t channels, size_t output_width, const uint8_t **input, const void *weights, int32_t *buffer, uint8_t *output, size_t input_stride, size_t output_increment, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gavgpool_up_ukernel_function) (size_t m, size_t n, const uint8_t *x, size_t x_stride, const uint8_t *zero, uint8_t *y, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8gavgpool_mp_ukernel_function) (size_t m, size_t n, const uint8_t *x, size_t x_stride, const uint8_t *zero, int32_t *buffer, uint8_t *y, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8avgpool_up_ukernel_function) (size_t n, size_t ks, size_t kc, const uint8_t **x, const uint8_t *zero, uint8_t *y, size_t x_increment, size_t y_increment, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* | pytorch_q8avgpool_mp_ukernel_function) (size_t n, size_t ks, size_t kc, const uint8_t **x, const uint8_t *zero, int32_t *buffer, uint8_t *y, size_t x_increment, size_t y_increment, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* | pytorch_u8maxpool_ukernel_function) (size_t n, size_t ks, size_t kc, const uint8_t **x, uint8_t *y, size_t x_increment, size_t y_increment, const union pytorch_qnnp_u8_clamping_params *params) |
| typedef void(* | pytorch_u8clamp_ukernel_function) (size_t n, const uint8_t *x, uint8_t *y, const union pytorch_qnnp_u8_clamping_params *params) |
| typedef uint8_t(* | pytorch_u8rmax_ukernel_function) (size_t n, const uint8_t *x) |
| typedef void(* | pytorch_u8lut32norm_ukernel_function) (size_t n, const uint8_t *x, const uint32_t *t, uint8_t *y) |
| typedef void(* | pytorch_q8vadd_ukernel_function) (size_t n, const uint8_t *a, const uint8_t *b, uint8_t *y, const union pytorch_qnnp_add_quantization_params *quantization_params) |
Variables | |
| struct pytorch_qnnp_parameters | pytorch_qnnp_params |
| typedef void(* pytorch_hgemm_ukernel_function) (size_t mr, size_t nr, size_t k, const void *a, size_t a_stride, const void *w, void *c, size_t c_stride, const struct pytorch_qnnp_fp16_clamping_params *clamping_params) |
| typedef void(* pytorch_q8avgpool_mp_ukernel_function) (size_t n, size_t ks, size_t kc, const uint8_t **x, const uint8_t *zero, int32_t *buffer, uint8_t *y, size_t x_increment, size_t y_increment, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* pytorch_q8avgpool_up_ukernel_function) (size_t n, size_t ks, size_t kc, const uint8_t **x, const uint8_t *zero, uint8_t *y, size_t x_increment, size_t y_increment, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* pytorch_q8conv_ukernel_function) (size_t mr, size_t nr, size_t kc, size_t ks, const uint8_t **a, const void *w, uint8_t *c, size_t c_stride, size_t output_channel_index, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* pytorch_q8dwconv_mp_ukernel_function) (size_t channels, size_t output_width, const uint8_t **input, const void *weights, int32_t *buffer, uint8_t *output, size_t input_stride, size_t output_increment, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* pytorch_q8dwconv_up_ukernel_function) (size_t channels, size_t output_width, const uint8_t **input, const void *weights, uint8_t *output, size_t input_stride, size_t output_increment, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gavgpool_mp_ukernel_function) (size_t m, size_t n, const uint8_t *x, size_t x_stride, const uint8_t *zero, int32_t *buffer, uint8_t *y, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gavgpool_up_ukernel_function) (size_t m, size_t n, const uint8_t *x, size_t x_stride, const uint8_t *zero, uint8_t *y, const union pytorch_qnnp_avgpool_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gemm_dq_sparse_packedA_ukernel_function) (size_t mr, size_t nr, const uint8_t *a_packed, const uint8_t *packed_w, const uint32_t *w_row_ptr, const uint32_t *w_block_ids_ptr, const float *bias, float *c, size_t c_stride, size_t output_channel_index, const struct pytorch_qnnp_conv_dynamic_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gemm_dq_sparse_ukernel_function) (size_t mr, size_t nr, const uint8_t *a, size_t a_stride, const uint8_t *packed_w, const uint32_t *w_row_ptr, const uint32_t *w_block_ids_ptr, const float *bias, float *c, size_t c_stride, size_t output_channel_index, const struct pytorch_qnnp_conv_dynamic_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gemm_dq_ukernel_function) (size_t mr, size_t nr, size_t k, const uint8_t *a, size_t a_stride, const void *w, const float *bias, float *c, size_t c_stride, size_t output_channel_index, const struct pytorch_qnnp_conv_dynamic_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gemm_sparse_packA_ukernel_function) (const size_t mr, const size_t K, const uint8_t *a, const size_t a_stride, uint8_t *a_packed) |
| typedef void(* pytorch_q8gemm_ukernel_function) (size_t mr, size_t nr, size_t k, const uint8_t *a, size_t a_stride, const void *w, uint8_t *c, size_t c_stride, size_t output_channel_index, const union pytorch_qnnp_conv_quantization_params *quantization_params) |
| typedef void(* pytorch_q8gemm_xzp_ukernel_function) (size_t mr, size_t nr, size_t k, const uint8_t *a, size_t a_stride, const int32_t *a_sum, const void *w, uint8_t *c, size_t c_stride, const union pytorch_qnnp_q31_requantization_params *requantization_params) |
| typedef void(* pytorch_q8sum_rows_ukernel_function) (const uint8_t *a, size_t m, size_t k, size_t stride, int32_t multiplier, int32_t *sums) |
| typedef void(* pytorch_q8vadd_ukernel_function) (size_t n, const uint8_t *a, const uint8_t *b, uint8_t *y, const union pytorch_qnnp_add_quantization_params *quantization_params) |
| typedef void(* pytorch_sconv_ukernel_function) (size_t mr, size_t nr, size_t kc, size_t ks, const float **a, const float *w, float *c, size_t c_stride, const struct pytorch_qnnp_fp32_clamping_params *clamping_params) |
| typedef void(* pytorch_sgemm_ukernel_function) (size_t mr, size_t nr, size_t k, const float *a, size_t a_stride, const float *w, float *c, size_t c_stride, const struct pytorch_qnnp_fp32_clamping_params *clamping_params) |
| typedef void(* pytorch_u8clamp_ukernel_function) (size_t n, const uint8_t *x, uint8_t *y, const union pytorch_qnnp_u8_clamping_params *params) |
| typedef void(* pytorch_u8lut32norm_ukernel_function) (size_t n, const uint8_t *x, const uint32_t *t, uint8_t *y) |
| typedef void(* pytorch_u8maxpool_ukernel_function) (size_t n, size_t ks, size_t kc, const uint8_t **x, uint8_t *y, size_t x_increment, size_t y_increment, const union pytorch_qnnp_u8_clamping_params *params) |
| typedef uint8_t(* pytorch_u8rmax_ukernel_function) (size_t n, const uint8_t *x) |
| typedef void(* pytorch_x8lut_ukernel_function) (size_t n, const uint8_t *x, const uint8_t *t, uint8_t *y) |
| typedef void(* pytorch_xzipc_ukernel_function) (size_t n, const void *x, void *y) |
| typedef void( * pytorch_xzipv_ukernel_function) (size_t n, size_t m, const void *x, void *y) |
|
extern |
Definition at line 45 of file init.c.
Referenced by init(), qnnpack::PackBMatrix::PackBMatrix(), qnnpack::PrePackConvWeights::PrePackConvWeights(), pytorch_qnnp_create_add_nc_q8(), pytorch_qnnp_create_average_pooling2d_nhwc_q8(), pytorch_qnnp_create_channel_shuffle_nc_x8(), pytorch_qnnp_create_clamp_nc_u8(), pytorch_qnnp_create_convolution2d_nhwc_q8(), pytorch_qnnp_create_deconvolution2d_nhwc_q8(), pytorch_qnnp_create_fully_connected_nc_q8(), pytorch_qnnp_create_fully_connected_sparse_dq_nc_q8(), pytorch_qnnp_create_global_average_pooling_nwc_q8(), pytorch_qnnp_create_hardsigmoid_nc_q8(), pytorch_qnnp_create_hardswish_nc_q8(), pytorch_qnnp_create_leaky_relu_nc_q8(), pytorch_qnnp_create_max_pooling2d_nhwc_u8(), pytorch_qnnp_create_sigmoid_nc_q8(), pytorch_qnnp_create_softargmax_nc_q8(), pytorch_qnnp_create_tanh_nc_q8(), pytorch_qnnp_initialize(), pytorch_qnnp_run_operator(), pytorch_qnnp_setup_add_nc_q8(), pytorch_qnnp_setup_average_pooling2d_nhwc_q8(), pytorch_qnnp_setup_channel_shuffle_nc_x8(), pytorch_qnnp_setup_clamp_nc_u8(), pytorch_qnnp_setup_convolution2d_nhwc_q8(), pytorch_qnnp_setup_deconvolution2d_nhwc_q8(), pytorch_qnnp_setup_fully_connected_nc_q8(), pytorch_qnnp_setup_fully_connected_sparse_dq_nc_q8(), pytorch_qnnp_setup_global_average_pooling_nwc_q8(), pytorch_qnnp_setup_hardsigmoid_nc_q8(), pytorch_qnnp_setup_hardswish_nc_q8(), pytorch_qnnp_setup_leaky_relu_nc_q8(), pytorch_qnnp_setup_max_pooling2d_nhwc_u8(), pytorch_qnnp_setup_sigmoid_nc_q8(), pytorch_qnnp_setup_softargmax_nc_q8(), pytorch_qnnp_setup_tanh_nc_q8(), qnnpack::qnnpackConv(), qnnpack::qnnpackDeConv(), qnnpack::qnnpackLinear(), and qnnpack::qnnpackLinearDynamic().
1.9.2