diff --git a/mace/ops/eltwise.cc b/mace/ops/eltwise.cc index bb7532cc3f79cd461fb5e891476955038c0c4bae..d8520dcb37d578c41c2536a3e66420ec72b416b2 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(); 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();