From d8b13dee5e656bdaa88652e1c8d7ef1cb41d373a Mon Sep 17 00:00:00 2001
From: sweetsky0901 <work@yq01-idl-gpu-online20.yq01.baidu.com>
Date: Fri, 22 Dec 2017 15:10:28 +0800
Subject: [PATCH] add norm_op for ssd(cross channel norm)

---
 paddle/operators/norm_op.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/operators/norm_op.cc b/paddle/operators/norm_op.cc
index 3835da630..d23805da8 100644
--- a/paddle/operators/norm_op.cc
+++ b/paddle/operators/norm_op.cc
@@ -68,6 +68,9 @@ class NormOp : public framework::OperatorWithKernel {
     PADDLE_ENFORCE(ctx->HasInput("X"),
                    "Input(X) of NormOp"
                    "should not be null.");
+    PADDLE_ENFORCE(ctx->HasInput("Scale"),
+                   "Input(Scale) of NormOp"
+                   "should not be null.");
     PADDLE_ENFORCE(ctx->HasOutput("Out"),
                    "Output(Out) of NormOp should not be null.");
     auto in_x_dims = ctx->GetInputDim("X");
-- 
GitLab