From ea00270fe8cf1130d0d6741811acaccaad50c0e3 Mon Sep 17 00:00:00 2001 From: Yihua Xu Date: Mon, 3 Dec 2018 14:16:07 +0800 Subject: [PATCH] Remove the dims checking when the dim is 3 (test=develop) --- paddle/fluid/operators/activation_mkldnn_op.cc | 3 --- 1 file changed, 3 deletions(-) diff --git a/paddle/fluid/operators/activation_mkldnn_op.cc b/paddle/fluid/operators/activation_mkldnn_op.cc index 64649b1a5..7fa81e185 100644 --- a/paddle/fluid/operators/activation_mkldnn_op.cc +++ b/paddle/fluid/operators/activation_mkldnn_op.cc @@ -100,9 +100,6 @@ void eltwise_forward(const framework::ExecutionContext &ctx, const T *x_data = x->data(); T *y_data = y->mutable_data(ctx.GetPlace()); - PADDLE_ENFORCE(x->dims().size() == 2 || x->dims().size() == 4, - "Input dim must be with 2 or 4"); - std::vector src_tz = framework::vectorize2int(x->dims()); auto src_format = -- GitLab