未验证 提交 181b1f5a 编写于 作者: P Pei Yang 提交者: GitHub

adjust minimum trt version for hard_sigmoid converter to 5130. test=develop (#24746)

上级 79caed66
...@@ -25,7 +25,7 @@ class HardSigmoidOpConverter : public OpConverter { ...@@ -25,7 +25,7 @@ class HardSigmoidOpConverter : public OpConverter {
public: public:
void operator()(const framework::proto::OpDesc& op, void operator()(const framework::proto::OpDesc& op,
const framework::Scope& scope, bool test_mode) override { const framework::Scope& scope, bool test_mode) override {
#if IS_TRT_VERSION_GE(5000) #if IS_TRT_VERSION_GE(5130)
VLOG(3) << "convert a fluid HardSigmoid op to tensorrt IActivationLayer " VLOG(3) << "convert a fluid HardSigmoid op to tensorrt IActivationLayer "
"layer without bias"; "layer without bias";
framework::OpDesc op_desc(op, nullptr); framework::OpDesc op_desc(op, nullptr);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册