自定义op
文章平均质量分 58
Ehrich wen
人工智能、算法集成、TensorRT、OpenVINO、TNN&MNN、C++后端开发
展开
-
TensorRT 3D反卷积不支持非对称padding
转trt模型时报错: ASSERT(begPadding == endPadding && “TensorRT does not support asymmetrical padding for 3D deconvolutions!”, ErrorCode::kUNSUPPORTED_NODE);原因:TensorRT目前还不支持3D反卷积非对称padding解决方法:找到源码TensorRT/parsers/onnx/builtin_op_importers.cpp文件,找到Con原创 2022-01-26 16:32:31 · 3303 阅读 · 0 评论 -
TensorRT动态卷积转置(ConvTranspose)自定义op(动态反卷积/权重动态反卷积)
随着算法模型的不断演变,常规的模型结构已经不能满足算法人员的需求,于是衍生出形色各异的op,比如动态卷积转置,即前向传播过程中,weight也会随着输入的不同而发生改变。声明:所谓动态卷积转置指在前向传播过程中weight发生变化此处的动态卷积转置只是笔者对该op的一种称呼本文ConvTranspose只支持2D目前发现只在TensorRT中存在动态的问题,其他如onnx、OpenVINO不存在该问题动态卷积转置大概长下面这个样子。其中input是输入,weight也是输入。weight根据前原创 2022-01-10 11:39:50 · 1019 阅读 · 0 评论