diff options
Diffstat (limited to 'src/xnnpack/microfnptr.h')
-rw-r--r-- | src/xnnpack/microfnptr.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/xnnpack/microfnptr.h b/src/xnnpack/microfnptr.h index 71396f3bd..401fb4861 100644 --- a/src/xnnpack/microfnptr.h +++ b/src/xnnpack/microfnptr.h @@ -1331,6 +1331,14 @@ typedef void (*xnn_f32_vsqrt_ukernel_function)( float* output, const union xnn_f32_sqrt_params* params); +// VSQRTSHIFT: Vector SQuare RooT and SHIFT elementwise + +typedef void (*xnn_u64_u32_vsqrtshift_ukernel_function)( + size_t batch, + const uint64_t* input, + uint32_t* output, + const union xnn_u64_u32_sqrtshift_params* params); + // LUT: vector LookUp Table elementwise typedef void (*xnn_x8_lut_ukernel_function)( @@ -2013,6 +2021,10 @@ typedef size_t (*xnn_init_f16_sqrt_params_fn)( typedef size_t (*xnn_init_f32_sqrt_params_fn)( union xnn_f32_sqrt_params params[XNN_MIN_ELEMENTS(1)]); +typedef size_t (*xnn_init_u64_u32_sqrtshift_params_fn)( + union xnn_u64_u32_sqrtshift_params params[XNN_MIN_ELEMENTS(1)], + uint32_t shift); + typedef void (*xnn_init_qc8_scale_params_fn)( size_t channels, size_t channels_tile, |