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 9d80edd commit dafd6d0Copy full SHA for dafd6d0
paddle/fluid/inference/tensorrt/convert/ut_helper.h
@@ -163,6 +163,9 @@ class TRTConvertValidation {
163
}
164
165
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.
169
void Execute(int batch_size,
170
std::unordered_set<std::string> neglected_output = {}) {
171
// Execute Fluid Op
0 commit comments