24.02.1
|
Go to the documentation of this file.
24 #ifndef SRC_CORE_NEON_KERNELS_ROIALIGN_LIST_H
25 #define SRC_CORE_NEON_KERNELS_ROIALIGN_LIST_H
30 #define DECLARE_ROIALIGN_KERNEL(func_name) \
31 void func_name(const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, \
32 const Window &window, const ThreadInfo &info)
37 #undef DECLARE_ROIALIGN_KERNEL
void neon_fp32_roialign(const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, const Window &window, const ThreadInfo &info)
#define DECLARE_ROIALIGN_KERNEL(func_name)
void neon_qs8_roialign(const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, const Window &window, const ThreadInfo &info)
void neon_qu8_roialign(const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, const Window &window, const ThreadInfo &info)
void neon_fp16_roialign(const ITensor *input, ITensor *output, const ITensor *rois, ROIPoolingLayerInfo pool_info, const Window &window, const ThreadInfo &info)
Copyright (c) 2017-2024 Arm Limited.