| 113 | } |
| 114 | |
| 115 | void LogicalOps::InitRegistry(void) |
| 116 | { |
| 117 | #ifdef CONFIG_KERNEL_FP32 |
| 118 | kernel_registry.Register(( ref_logical_t )ref_logical_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 119 | kernel_registry.Register(( ref_logical_t )ref_logical_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 120 | #endif |
| 121 | |
| 122 | // #ifdef CONFIG_KERNEL_FP16 |
| 123 | // kernel_registry.Register(( ref_logical_t )ref_logical_fp16, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP16); |
| 124 | // kernel_registry.Register(( ref_logical_t )ref_logical_fp16, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP16); |
| 125 | // #endif |
| 126 | // #ifdef CONFIG_KERNEL_INT8 |
| 127 | // kernel_registry.Register(( ref_logical_t )ref_logical_int8, TENGINE_LAYOUT_NCHW, TENGINE_DT_INT8); |
| 128 | // kernel_registry.Register(( ref_logical_t )ref_logical_int8, TENGINE_LAYOUT_NHWC, TENGINE_DT_INT8); |
| 129 | // #endif |
| 130 | |
| 131 | // #ifdef CONFIG_KERNEL_UINT8 |
| 132 | // kernel_registry.Register(( ref_logical_t )ref_logical_uint8, TENGINE_LAYOUT_NCHW, TENGINE_DT_UINT8); |
| 133 | // kernel_registry.Register(( ref_logical_t )ref_logical_uint8, TENGINE_LAYOUT_NHWC, TENGINE_DT_UINT8); |
| 134 | // #endif |
| 135 | } |
| 136 | |
| 137 | NodeOps* SelectFunc(const CPUInfo* info, Node* node) |
| 138 | { |