忧心溪流

文章
5
资源
0
加入时间
2年10月17天

caffe Reshape层TensorRT IPlugin代码实现

目前TensorRT并不支持caffe的Reshape层,故在用TensorRT加速caffe网络时,Reshape需要通过TensorRT的plugin插件实现。以下是Reshape层的Plugin源码:#include <cuda_runtime.h>#include "NvInferPlugin.h"#define CHECK(status)\ {\ ...

都是显存惹的祸

  笔者的机器升级是在2001年6月,整机配置如下:磐英3SPA,CII667,HY12