提交 c62b41fa 编写于 作者: wayen820's avatar wayen820 提交者: Bin Li

fix eltwise op bug when input1.size()==input0.size() (#252)

上级 7cb18a70
......@@ -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>();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册