Skip to content

Commit

Permalink
remove SetTensorDynamicRange in softmax (PaddlePaddle#57538)
Browse files Browse the repository at this point in the history
  • Loading branch information
zhoutianzi666 authored and iosmers committed Sep 21, 2023
1 parent 2a52a8c commit aff96e3
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions paddle/fluid/inference/tensorrt/convert/softmax_op.cc
Original file line number Diff line number Diff line change
Expand Up @@ -84,8 +84,6 @@ class SoftMaxOpConverter : public OpConverter {
}
layer->setAxes(1 << axes);

// The trt will not run int for softmax.
engine_->SetTensorDynamicRange(input1, 1.0);
auto output_name = op_desc.Output("Out")[0];

// support 0 or 1 dims input
Expand Down

0 comments on commit aff96e3

Please sign in to comment.