From dafd6d062cc36d2e0137eac5fe8ec5bdd0ae4d83 Mon Sep 17 00:00:00 2001
From: nhzlx <zlx_hg@163.com>
Date: Tue, 21 Aug 2018 02:53:00 +0000
Subject: [PATCH] Merge pull request from #12786 from
 NHZlX:add_batch_norm_trt_converter

---
 paddle/fluid/inference/tensorrt/convert/ut_helper.h | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/fluid/inference/tensorrt/convert/ut_helper.h b/paddle/fluid/inference/tensorrt/convert/ut_helper.h
index bf740d38f..0a6f171fc 100644
--- a/paddle/fluid/inference/tensorrt/convert/ut_helper.h
+++ b/paddle/fluid/inference/tensorrt/convert/ut_helper.h
@@ -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
-- 
GitLab