diff options
Diffstat (limited to 'src/xnnpack/compute.h')
| -rw-r--r-- | src/xnnpack/compute.h | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/src/xnnpack/compute.h b/src/xnnpack/compute.h index db6ff72ce..7370b143f 100644 --- a/src/xnnpack/compute.h +++ b/src/xnnpack/compute.h @@ -839,6 +839,18 @@ struct elementwise_binary_context { }; #ifndef __cplusplus + XNN_PRIVATE void xnn_compute_elementwise_binary_1d( + const struct elementwise_binary_context context[restrict XNN_MIN_ELEMENTS(1)], + size_t i); + XNN_PRIVATE void xnn_compute_elementwise_binary_2d( + const struct elementwise_binary_context context[restrict XNN_MIN_ELEMENTS(1)], + size_t i, size_t j); + XNN_PRIVATE void xnn_compute_elementwise_binary_3d( + const struct elementwise_binary_context context[restrict XNN_MIN_ELEMENTS(1)], + size_t i, size_t j, size_t k); + XNN_PRIVATE void xnn_compute_elementwise_binary_4d( + const struct elementwise_binary_context context[restrict XNN_MIN_ELEMENTS(1)], + size_t i, size_t j, size_t k, size_t l); XNN_PRIVATE void xnn_compute_elementwise_binary_5d( const struct elementwise_binary_context context[restrict XNN_MIN_ELEMENTS(1)], size_t i, size_t j, size_t k, size_t l, size_t m); |