Skip to content

Commit dafd6d0

Browse files
NHZlXPaddle CI
authored andcommitted
Merge pull request from #12786 from NHZlX:add_batch_norm_trt_converter
1 parent 9d80edd commit dafd6d0

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
@@ -163,6 +163,9 @@ class TRTConvertValidation {
163163
}
164164
}
165165

166+
// We use the set 'neglected_output' here, because some Ops like batch norm,
167+
// the outputs specified in the op des are only used during training,
168+
// so we should neglect those output during inference.
166169
void Execute(int batch_size,
167170
std::unordered_set<std::string> neglected_output = {}) {
168171
// Execute Fluid Op

0 commit comments

Comments
 (0)