1 // Copyright 2020 Google LLC
2 //
3 // This source code is licensed under the BSD-style license found in the
4 // LICENSE file in the root directory of this source tree.
5 
6 #pragma once
7 
8 #include <stddef.h>
9 #include <stdint.h>
10 
11 #include <xnnpack/params.h>
12 #include <xnnpack/common.h>
13 
14 #ifdef __cplusplus
15 extern "C" {
16 #endif
17 
18 
19 #define DECLARE_FILL_UKERNEL_FUNCTION(fn_name) \
20   XNN_INTERNAL void fn_name(                   \
21     size_t kernel_elements,                    \
22     size_t channels,                           \
23     uint32_t* output,                          \
24     size_t output_stride,                      \
25     const uint32_t* fill_value);
26 
27 DECLARE_FILL_UKERNEL_FUNCTION(xnn_x32_fill_ukernel__sse)
28 DECLARE_FILL_UKERNEL_FUNCTION(xnn_x32_fill_ukernel__neon)
29 DECLARE_FILL_UKERNEL_FUNCTION(xnn_x32_fill_ukernel__wasmsimd)
30 DECLARE_FILL_UKERNEL_FUNCTION(xnn_x32_fill_ukernel__scalar_float)
31 DECLARE_FILL_UKERNEL_FUNCTION(xnn_x32_fill_ukernel__scalar_int)
32 
33 
34 #ifdef __cplusplus
35 }  // extern "C"
36 #endif
37