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 @@ -24,7 +24,7 @@ class ReluOpConverter : public OpConverter {
24
24
void operator ()(const framework::proto::OpDesc& op) override {
25
25
// Here the two nullptr looks strange, that's because the
26
26
// framework::OpDesc's constructor is strange.
27
- framework::OpDesc op_desc (op, nullptr , nullptr );
27
+ framework::OpDesc op_desc (op, nullptr );
28
28
LOG (INFO) << " convert a fluid relu op to tensorrt activation layer whose "
29
29
" type is Relu" ;
30
30
const nvinfer1::ITensor* input_tensor =
Original file line number Diff line number Diff line change @@ -27,7 +27,7 @@ class MulOpConverter : public OpConverter {
27
27
void operator ()(const framework::proto::OpDesc& op) override {
28
28
VLOG (4 ) << " convert a fluid mul op to tensorrt fc layer without bias" ;
29
29
30
- framework::OpDesc op_desc (op, nullptr , nullptr );
30
+ framework::OpDesc op_desc (op, nullptr );
31
31
// Declare inputs
32
32
auto * input1 = engine_->GetITensor (op_desc.Input (" X" )[0 ]);
33
33
auto * input2 = engine_->GetITensor (op_desc.Input (" Y" )[0 ]);
You can’t perform that action at this time.
0 commit comments