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

v1.8
Pei Yang 5 years ago committed by GitHub
parent 79caed6667
commit 181b1f5a30
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -25,7 +25,7 @@ class HardSigmoidOpConverter : public OpConverter {
public:
void operator()(const framework::proto::OpDesc& op,
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 "
"layer without bias";
framework::OpDesc op_desc(op, nullptr);

Loading…
Cancel
Save