提交 657cd06d 编写于 作者: S SunAhong1993

fix the slice

上级 3ca4aed5
......@@ -912,7 +912,9 @@ class TFOpMapper(OpMapper):
# shape=shape)
# inputs['shape'] = reshape_name
# inputs.pop('shape')
for i, s in enumerate(shape):
if s < 0:
shape[i] = 32767
program.add_layer(
kernel="fluid.layers.slice",
inputs=inputs,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册