提交 dafd6d06 编写于 作者: N nhzlx 提交者: Paddle CI

Merge pull request from #12786 from NHZlX:add_batch_norm_trt_converter

上级 9d80edd6
......@@ -163,6 +163,9 @@ class TRTConvertValidation {
}
}
// We use the set 'neglected_output' here, because some Ops like batch norm,
// the outputs specified in the op des are only used during training,
// so we should neglect those output during inference.
void Execute(int batch_size,
std::unordered_set<std::string> neglected_output = {}) {
// Execute Fluid Op
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册