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)  

4x8-neon.c File Reference
#include <arm_neon.h>
#include <qnnpack/q8conv.h>
#include <requantization/runtime-neon.h>
Include dependency graph for 4x8-neon.c:

Go to the source code of this file.


void pytorch_q8conv_ukernel_4x8__neon (size_t mr, size_t nr, size_t kc, size_t ks, const uint8_t **restrict a, const void *restrict w, uint8_t *restrict c, size_t c_stride, size_t output_channel_index, const union pytorch_qnnp_conv_quantization_params quantization_params[restrict static 1])

Function Documentation

◆ pytorch_q8conv_ukernel_4x8__neon()

void pytorch_q8conv_ukernel_4x8__neon ( size_t  mr,
size_t  nr,
size_t  kc,
size_t  ks,
const uint8_t **restrict  a,
const void *restrict  w,
uint8_t *restrict  c,
size_t  c_stride,
size_t  output_channel_index,
const union pytorch_qnnp_conv_quantization_params  quantization_params[restrict static 1]