提交 ce92c536 编写于 作者: S SunAhong1993

fix the paddle.slice

上级 78d6804a
......@@ -958,8 +958,10 @@ class OpSet9():
starts_value[idx] = val_x.out_shapes[0][axes[idx]] - 1
ends_value[idx] = val_x.out_shapes[0][axes[idx]]
starts_value[idx] = val_x.out_shapes[0][axes[idx]] - 1
elif ends_value[idx] >= 2**31 - 1:
elif ends_value[idx] == 2**31 - 1:
ends_value[idx] = node.out_shapes[0][axes[idx]] + 1
elif ends_value[idx] > 2**31 - 1:
ends_value[idx] = 2**31 - 1
layer_attrs = {
"axes": axes,
"starts": starts_value,
......
......@@ -922,8 +922,10 @@ class OpSet9():
starts_value[idx] = val_x.out_shapes[0][axes[idx]] - 1
ends_value[idx] = val_x.out_shapes[0][axes[idx]]
starts_value[idx] = val_x.out_shapes[0][axes[idx]] - 1
elif ends_value[idx] >= 2**31 - 1:
elif ends_value[idx] == 2**31 - 1:
ends_value[idx] = node.out_shapes[0][axes[idx]] + 1
elif ends_value[idx] > 2**31 - 1:
ends_value[idx] = 2**31 - 1
layer_attrs = {
"axes": axes,
"starts": starts_value,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册