提交 bfc20238 编写于 作者: C channingss

little modify

上级 d7f2b116
......@@ -199,7 +199,14 @@ class TFOpMapper(OpMapper):
shape = [1]
if value == float('inf'):
value = "float('inf')"
initializer = "Constant({})".format(value)
program.add_layer(
kernel="fluid.layers.fill_constant",
inputs={},
outputs=[node.name],
dtype=string(dtype),
shape=[1],
value=value)
return
program.parameters[node.name] = node.value
program.add_layer(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册