Skip to content

Commit 1bf9d9e

Browse files
committed
fix comments
1 parent 324dd16 commit 1bf9d9e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

paddle/fluid/inference/tensorrt/convert/ut_helper.h

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,9 @@ class TRTConvertValidation {
177177
ASSERT_FALSE(op_desc_->OutputArgumentNames().empty());
178178
const size_t output_space_size = 3000;
179179
for (const auto& output : op_desc_->OutputArgumentNames()) {
180+
// Some Ops like batch norm, the output specified
181+
// in the op des is only used during training, so we should
182+
// neglect those ouput during inference.
180183
if (neglected_output.count(output)) continue;
181184
std::vector<float> fluid_out;
182185
std::vector<float> trt_out(output_space_size);

0 commit comments

Comments
 (0)