File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed
paddle/fluid/inference/tensorrt/convert Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -59,7 +59,7 @@ class FcOpConverter : public OpConverter {
59
59
const framework::Scope& scope) override {
60
60
VLOG (4 ) << " convert a fluid fc op to tensorrt fc layer without bias" ;
61
61
62
- framework::OpDesc op_desc (op, nullptr , nullptr );
62
+ framework::OpDesc op_desc (op, nullptr );
63
63
PADDLE_ENFORCE_EQ (op_desc.Input (" X" ).size (), 1 );
64
64
PADDLE_ENFORCE_EQ (op_desc.Input (" Y" ).size (), 1 ); // Y is a weight
65
65
PADDLE_ENFORCE_EQ (op_desc.Output (" Out" ).size (), 1 );
Original file line number Diff line number Diff line change @@ -40,7 +40,7 @@ class OpConverter {
40
40
void ConvertOp (const framework::proto::OpDesc& op,
41
41
const std::unordered_set<std::string>& parameters,
42
42
const framework::Scope& scope, TensorRTEngine* engine) {
43
- framework::OpDesc op_desc (op, nullptr , nullptr );
43
+ framework::OpDesc op_desc (op, nullptr );
44
44
45
45
OpConverter* it{nullptr };
46
46
You can’t perform that action at this time.
0 commit comments