operators_maps for all ops

zjhellofss / KuiperInfer

校招、秋招、春招、实习好项目！带你从零实现一个高性能的深度学习推理库，支持大模型 llama2 、Unet、Yolov5、Resnet等模型的推理。Implement a high-performance deep learning inference library step by step

MIT License

2.59k stars 295 forks source link

Closed Tigerrr07 closed 1 year ago

Tigerrr07 commented 1 year ago

所有的operators以<name, op>的map存储，构建图关系时，不用再遍历整个operators，以及std::find(name)，根据output_names直接插入map中存储的节点。
这样去掉了input_operators_maps和ioutput_operators_maps，需要他们时，用op->type进行区分
修改后运行了yolo_test，能够正确识别