MST-plus-plus-TensorRT
MST-plus-plus-TensorRT copied to clipboard
请问动态 shape 下的空间如何申请呢?
请问,大佬知道 dynamic shape 下的 getWorkspaceSize 如何写吗?静态的很好判断,但目前个人自定义的算子卡在了这里,还望大佬指点!
- https://docs.nvidia.com/deeplearning/tensorrt/developer-guide/index.html#build_engine_c
- 我不确定您说的·getWorkspaceSize`是不是这个,我一般的做法是默认值,如果不行会在默认值的基础上增加,直到可以正常导出