From 1bf9d9e90cc1c9e7ab8a5550db3eb99dd8433dfe Mon Sep 17 00:00:00 2001
From: nhzlx <zlx_hg@163.com>
Date: Mon, 20 Aug 2018 07:20:21 +0000
Subject: [PATCH] fix comments

---
 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 bf740d38f0f..833f35167b9 100644
--- a/paddle/fluid/inference/tensorrt/convert/ut_helper.h
+++ b/paddle/fluid/inference/tensorrt/convert/ut_helper.h
@@ -177,6 +177,9 @@ class TRTConvertValidation {
     ASSERT_FALSE(op_desc_->OutputArgumentNames().empty());
     const size_t output_space_size = 3000;
     for (const auto& output : op_desc_->OutputArgumentNames()) {
+      // Some Ops like batch norm, the output specified
+      // in the op des is only used during training, so we should
+      // neglect those ouput during inference.
       if (neglected_output.count(output)) continue;
       std::vector<float> fluid_out;
       std::vector<float> trt_out(output_space_size);
-- 
GitLab