From c62b41fa4ac6cc54bae97291ced4aaa26e292b83 Mon Sep 17 00:00:00 2001
From: wayen820 <1053988516@qq.com>
Date: Fri, 9 Nov 2018 11:28:59 +0800
Subject: [PATCH] fix eltwise op bug when input1.size()==input0.size() (#252)

---
 mace/ops/eltwise.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/ops/eltwise.cc b/mace/ops/eltwise.cc
index bb7532cc..d8520dcb 100644
--- a/mace/ops/eltwise.cc
+++ b/mace/ops/eltwise.cc
@@ -865,7 +865,7 @@ class EltwiseOp : public Operation {
     const T *input1_ptr = input1->data<T>();
 
     if (data_format_ == NCHW && input1->dim_size() > 0 &&
-        input1->size() < input0->size()) {
+        input1->size() <= input0->size()) {
       MACE_RETURN_IF_ERROR(output->ResizeLike(input0));
       Tensor::MappingGuard output_guard(output);
       DstType *output_ptr = output->mutable_data<DstType>();
-- 
GitLab