[operator] prototype/pad: fix unexpected output tensor dim 0 (#667)

adb37e7c · qinhj · GitHub · 0e90aaa0 · adb37e7c
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

source/operator/prototype/pad.c source/operator/prototype/pad.c +1 -1

未找到文件。
--- a/source/operator/prototype/pad.c
+++ b/source/operator/prototype/pad.c
@@ -51,7 +51,7 @@ static int infer_shape(ir_node_t* node)
    }
    else
    {
-        return 0;
+        return -1;
    }

    set_ir_tensor_shape(output, dims, input->dim_num);