| 73 | } |
| 74 | |
| 75 | void RoundOps::InitRegistry(void) |
| 76 | { |
| 77 | #ifdef CONFIG_KERNEL_FP32 |
| 78 | kernel_registry.Register(( ref_round_t )ref_round_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 79 | kernel_registry.Register(( ref_round_t )ref_round_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 80 | #endif |
| 81 | } |
| 82 | |
| 83 | NodeOps* SelectFunc(const CPUInfo* info, Node* node) |
| 84 | { |