| 352 | }; |
| 353 | |
| 354 | NodeOps* SelectFunc(const CPUInfo* cpu_info, Node* node) |
| 355 | { |
| 356 | Tensor* input = node->GetInputTensor(0); |
| 357 | const int data_type = input->GetDataType(); |
| 358 | const ExecAttr* exec_attr = any_cast<const ExecAttr*>(node->GetAttr(ATTR_EXEC_ATTR)); |
| 359 | if(data_type != TENGINE_DT_FP32 || exec_attr->graph_layout != TENGINE_LAYOUT_NCHW) |
| 360 | return nullptr; |
| 361 | |
| 362 | FMOps* ops = new FMOps(); |
| 363 | |
| 364 | int master_cpu = cpu_info->GetMasterCPU(); |
| 365 | |
| 366 | if(cpu_info->GetCPUModelString(master_cpu) == std::string("A72")) |
| 367 | ops->cpu_type = TYPE_A72; |
| 368 | else |
| 369 | ops->cpu_type = TYPE_A53; |
| 370 | |
| 371 | return ops; |
| 372 | } |
| 373 | |
| 374 | } // namespace FM_fast |
| 375 |
nothing calls this directly
no test coverage detected