提交 bef6da3e 编写于 作者: M mamingjie-China

support paddlepaddle 1.6.0

上级 96fb50ad
......@@ -278,6 +278,7 @@ class TFOpMapper(OpMapper):
'name': string(node.layer_name),
'append_batch_size': False
}
if shape[0] < 0:
self.batch_node = node
......
......@@ -1109,6 +1109,7 @@ class TFOpMapperNHWC(OpMapper):
else:
shape = self.decoder.infer_shape_tensor(shape)
attr = {"shape": shape, "min": 0.0, "max": 0.9999}
if shape[0] < 0:
input = self.batch_node
node.fluid_code.add_layer("uniform_random_batch_size_like",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册