diff options
Diffstat (limited to 'src/xnnpack')
| -rw-r--r-- | src/xnnpack/microfnptr.h | 6 | ||||
| -rw-r--r-- | src/xnnpack/vlshift.h | 6 |
2 files changed, 6 insertions, 6 deletions
diff --git a/src/xnnpack/microfnptr.h b/src/xnnpack/microfnptr.h index 6499eba53..feeb633d8 100644 --- a/src/xnnpack/microfnptr.h +++ b/src/xnnpack/microfnptr.h @@ -1656,10 +1656,10 @@ typedef void (*xnn_u32_filterbank_subtract_ukernel_function)( uint32_t* output); typedef void (*xnn_s16_vlshift_ukernel_function)( - size_t batch_size, + size_t batch, const int16_t* input, - uint32_t shift, - int16_t* output); + int16_t* output, + uint32_t shift); typedef void (*xnn_cs16_vsquareabs_ukernel_function)( size_t batch_size, diff --git a/src/xnnpack/vlshift.h b/src/xnnpack/vlshift.h index 31f3be488..cba428f67 100644 --- a/src/xnnpack/vlshift.h +++ b/src/xnnpack/vlshift.h @@ -17,10 +17,10 @@ extern "C" { #define DECLARE_S16_VLSHIFT_UKERNEL_FUNCTION(fn_name) \ XNN_INTERNAL void fn_name( \ - size_t batch_size, \ + size_t batch, \ const int16_t* input, \ - uint32_t shift, \ - int16_t* output); + int16_t* output, \ + uint32_t shift); DECLARE_S16_VLSHIFT_UKERNEL_FUNCTION(xnn_s16_vlshift_ukernel__neon_x8) |