| 139 | } |
| 140 | |
| 141 | void SigmoidOps::InitRegistry(void) |
| 142 | { |
| 143 | #ifdef CONFIG_KERNEL_FP32 |
| 144 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 145 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 146 | #endif |
| 147 | |
| 148 | #ifdef CONFIG_KERNEL_FP16 |
| 149 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_fp16, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP16); |
| 150 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_fp16, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP16); |
| 151 | #endif |
| 152 | #ifdef CONFIG_KERNEL_INT8 |
| 153 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_int8, TENGINE_LAYOUT_NCHW, TENGINE_DT_INT8); |
| 154 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_int8, TENGINE_LAYOUT_NHWC, TENGINE_DT_INT8); |
| 155 | #endif |
| 156 | |
| 157 | #ifdef CONFIG_KERNEL_UINT8 |
| 158 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_uint8, TENGINE_LAYOUT_NCHW, TENGINE_DT_UINT8); |
| 159 | kernel_registry.Register(( ref_sigmoid_t )ref_sigmoid_uint8, TENGINE_LAYOUT_NHWC, TENGINE_DT_UINT8); |
| 160 | #endif |
| 161 | } |
| 162 | |
| 163 | NodeOps* SelectFunc(const CPUInfo* info, Node* node) |
| 164 | { |