花费 8 ms
ONNXRuntime源码阅读(一)

在构建一个推理模型时(如NanoDet,一个目标检测模型),需要继承 BasicOrtHandler。BasicOrtHandler 的初始化函数中会调用 initialize_handler() 方 ...

Mon Mar 14 02:47:00 CST 2022 0 946
ONNXRuntime源码阅读(二)线程池

ONNXRuntime的线程池接口在Eigen线程池接口基础之上扩展而来(题外话:TensorFlow中的线程池同样是建立在Eigen线程池基础上),以下是线程池的继承关系,其中 ThreadPool ...

Mon Mar 14 19:50:00 CST 2022 0 735

 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM