MXNET部署TensorRT檢查OP是否存在

  MXNET模型進行部署時可以使用TensorRT,其過程通常爲將MXNET模型轉換爲ONNX,再通過TensorRT的ONNX解釋器轉換爲TensorRT的序列化文件。但此過程中,MXNET的OP不一定可以轉換爲ONNX的結果,而ONNX的OP也只有部分可以在TensorRT中實現。 TensorRT支持操作   可參照TensorRT官方文檔,這裏面除了介紹支持的OP操作外,還有關於支持的平
相關文章
相關標籤/搜索