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
Go to the documentation of this file.
1/*
2 * Copyright (c) Facebook, Inc. and its affiliates.
3 * All rights reserved.
4 *
5 * This source code is licensed under the BSD-style license found in the
6 * LICENSE file in the root directory of this source tree.
7 */
8
9#pragma once
10
11#include <stddef.h>
12#include <stdint.h>
13
14#include <qnnpack/common.h>
15#include <qnnpack/params.h>
16
17#ifdef __cplusplus
18extern "C" {
19#endif
20
21#define DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(fn_name) \
22 PYTORCH_QNNP_INTERNAL void fn_name( \
23 size_t mr, \
24 size_t nr, \
25 size_t kc, \
26 size_t ks, \
27 const uint8_t** a, \
28 const void* w, \
29 uint8_t* c, \
30 size_t c_stride, \
31 size_t output_channel_index, \
32 const union pytorch_qnnp_conv_quantization_params* quantization_params);
33
34DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(pytorch_q8conv_ukernel_4x8__neon)
35DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(pytorch_q8conv_ukernel_4x8__aarch32_neon)
36DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(pytorch_q8conv_ukernel_8x8__aarch64_neon)
37DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(pytorch_q8conv_ukernel_8x8__neon)
38DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(pytorch_q8conv_ukernel_4x4c2__sse2)
39
40#ifdef __cplusplus
41} /* extern "C" */
42#endif
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_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)
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)
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)
DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION
#define DECLARE_PYTORCH_Q8CONV_UKERNEL_FUNCTION(fn_name)
Definition: q8conv.h:21