| 57 | } |
| 58 | }; |
| 59 | void RefRoiPooling::InitRegistry(void) |
| 60 | { |
| 61 | #ifdef CONFIG_KERNEL_FP32 |
| 62 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 63 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 64 | #endif |
| 65 | #ifdef CONFIG_KERNEL_FP16 |
| 66 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_fp16, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP16); |
| 67 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_fp16, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP16); |
| 68 | #endif |
| 69 | #ifdef CONFIG_KERNEL_INT8 |
| 70 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_int8, TENGINE_LAYOUT_NCHW, TENGINE_DT_INT8); |
| 71 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_int8, TENGINE_LAYOUT_NHWC, TENGINE_DT_INT8); |
| 72 | #endif |
| 73 | #ifdef CONFIG_KERNEL_UINT8 |
| 74 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_uint8, TENGINE_LAYOUT_NCHW, TENGINE_DT_UINT8); |
| 75 | kernel_registry.Register(( ref_roipooling_kernel_t )ref_roipooling_uint8, TENGINE_LAYOUT_NHWC, TENGINE_DT_UINT8); |
| 76 | #endif |
| 77 | } |
| 78 | |
| 79 | bool RefRoiPooling::Prerun(Node* node) |
| 80 | { |