| 127 | } |
| 128 | |
| 129 | void RoialignOps::InitRegistry(void) |
| 130 | { |
| 131 | #ifdef CONFIG_KERNEL_FP32 |
| 132 | kernel_registry.Register(( ref_roialign_t )ref_roialign_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 133 | kernel_registry.Register(( ref_roialign_t )ref_roialign_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 134 | #endif |
| 135 | |
| 136 | #ifdef CONFIG_KERNEL_FP16 |
| 137 | kernel_registry.Register(( ref_roialign_t )ref_roialign_fp16, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP16); |
| 138 | kernel_registry.Register(( ref_roialign_t )ref_roialign_fp16, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP16); |
| 139 | #endif |
| 140 | #ifdef CONFIG_KERNEL_INT8 |
| 141 | kernel_registry.Register(( ref_roialign_t )ref_roialign_int8, TENGINE_LAYOUT_NCHW, TENGINE_DT_INT8); |
| 142 | kernel_registry.Register(( ref_roialign_t )ref_roialign_int8, TENGINE_LAYOUT_NHWC, TENGINE_DT_INT8); |
| 143 | #endif |
| 144 | |
| 145 | #ifdef CONFIG_KERNEL_UINT8 |
| 146 | kernel_registry.Register(( ref_roialign_t )ref_roialign_uint8, TENGINE_LAYOUT_NCHW, TENGINE_DT_UINT8); |
| 147 | kernel_registry.Register(( ref_roialign_t )ref_roialign_uint8, TENGINE_LAYOUT_NHWC, TENGINE_DT_UINT8); |
| 148 | #endif |
| 149 | |
| 150 | } |
| 151 | |
| 152 | NodeOps* SelectFunc(const CPUInfo* info, Node* node) |
| 153 | { |