| 188 | } |
| 189 | |
| 190 | void EltwiseOps::InitRegistry(void) |
| 191 | { |
| 192 | #ifdef CONFIG_KERNEL_FP32 |
| 193 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 194 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 195 | #endif |
| 196 | |
| 197 | #ifdef CONFIG_KERNEL_FP16 |
| 198 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_fp16, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP16); |
| 199 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_fp16, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP16); |
| 200 | #endif |
| 201 | #ifdef CONFIG_KERNEL_INT8 |
| 202 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_int8, TENGINE_LAYOUT_NCHW, TENGINE_DT_INT8); |
| 203 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_int8, TENGINE_LAYOUT_NHWC, TENGINE_DT_INT8); |
| 204 | #endif |
| 205 | |
| 206 | #ifdef CONFIG_KERNEL_UINT8 |
| 207 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_uint8, TENGINE_LAYOUT_NCHW, TENGINE_DT_UINT8); |
| 208 | kernel_registry.Register(( ref_eltwise_t )ref_eltwise_uint8, TENGINE_LAYOUT_NHWC, TENGINE_DT_UINT8); |
| 209 | #endif |
| 210 | } |
| 211 | |
| 212 | NodeOps* SelectFunc(const CPUInfo* info, Node* node) |
| 213 | { |