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)  

q8conv.h File Reference
#include <stddef.h>
#include <stdint.h>
#include <qnnpack/common.h>
#include <qnnpack/params.h>
Include dependency graph for q8conv.h:
This graph shows which files directly or indirectly include this file:

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)
 

Macro Definition Documentation

◆ DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION

#define DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION (   fn_name)
Value:
PYTORCH_QNNP_INTERNAL void fn_name( \
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 PYTORCH_QNNP_INTERNAL
Definition: common.h:59
const int * c

Definition at line 21 of file q8conv.h.

Function Documentation

◆ pytorch_q8conv_ukernel_4x4c2__sse2()

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_q8conv_ukernel_4x8__aarch32_neon()

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_q8conv_ukernel_4x8__neon()

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_q8conv_ukernel_8x8__aarch64_neon()

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_q8conv_ukernel_8x8__neon()

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 
)