| 87 | } |
| 88 | |
| 89 | void ComparisonOps::InitRegistry(void) |
| 90 | { |
| 91 | #ifdef CONFIG_KERNEL_FP32 |
| 92 | kernel_registry.Register(( ref_comparison_t )ref_comparison_fp32, TENGINE_LAYOUT_NCHW, TENGINE_DT_FP32); |
| 93 | kernel_registry.Register(( ref_comparison_t )ref_comparison_fp32, TENGINE_LAYOUT_NHWC, TENGINE_DT_FP32); |
| 94 | #endif |
| 95 | } |
| 96 | |
| 97 | NodeOps* SelectFunc(const CPUInfo* info, Node* node) |
| 98 | { |