提交 8f1f12ba 编写于 作者: J jiangjiajun

add dtype for range op

上级 0c37fdb5
......@@ -744,13 +744,12 @@ class TFOpMapperNHWC(OpMapper):
"start": start,
"end": limit,
"step": delta,
"dtype": string(dtype)
}
attr = {"dtype": string(node.dtype)}
node.fluid_code.add_layer("range",
inputs=inputs,
output=node,
param_attr=None)
param_attr=attr)
def Mean(self, node):
input = self.graph.get_node(node.layer.input[0], copy=True)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册