We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 324dd16 commit 1bf9d9eCopy full SHA for 1bf9d9e
paddle/fluid/inference/tensorrt/convert/ut_helper.h
@@ -177,6 +177,9 @@ class TRTConvertValidation {
177
ASSERT_FALSE(op_desc_->OutputArgumentNames().empty());
178
const size_t output_space_size = 3000;
179
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.
183
if (neglected_output.count(output)) continue;
184
std::vector<float> fluid_out;
185
std::vector<float> trt_out(output_space_size);
0 commit comments