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.
![]() ![]() |
Go to the source code of this file.
Macros | |
#define | DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(fn_name) |
Functions | |
PYTORCH_QNNP_INTERNAL void | pytorch_q8conv_ukernel_4x8__neon (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) |
PYTORCH_QNNP_INTERNAL void | pytorch_q8conv_ukernel_4x8__aarch32_neon (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) |
PYTORCH_QNNP_INTERNAL void | pytorch_q8conv_ukernel_8x8__aarch64_neon (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) |
PYTORCH_QNNP_INTERNAL void | pytorch_q8conv_ukernel_8x8__neon (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) |
PYTORCH_QNNP_INTERNAL void | pytorch_q8conv_ukernel_4x4c2__sse2 (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) |
#define DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION | ( | fn_name | ) |
PYTORCH_QNNP_INTERNAL void pytorch_q8conv_ukernel_4x4c2__sse2 | ( | 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 | ||
) |
PYTORCH_QNNP_INTERNAL void pytorch_q8conv_ukernel_4x8__aarch32_neon | ( | 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 | ||
) |
Referenced by init().
PYTORCH_QNNP_INTERNAL void pytorch_q8conv_ukernel_4x8__neon | ( | 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 | ||
) |
PYTORCH_QNNP_INTERNAL void pytorch_q8conv_ukernel_8x8__aarch64_neon | ( | 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 | ||
) |
Referenced by init().
PYTORCH_QNNP_INTERNAL void pytorch_q8conv_ukernel_8x8__neon | ( | 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 | ||
) |