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

add dtype for range op

上级 0c37fdb5
...@@ -744,13 +744,12 @@ class TFOpMapperNHWC(OpMapper): ...@@ -744,13 +744,12 @@ class TFOpMapperNHWC(OpMapper):
"start": start, "start": start,
"end": limit, "end": limit,
"step": delta, "step": delta,
"dtype": string(dtype)
} }
attr = {"dtype": string(node.dtype)} attr = {"dtype": string(node.dtype)}
node.fluid_code.add_layer("range", node.fluid_code.add_layer("range",
inputs=inputs, inputs=inputs,
output=node, output=node,
param_attr=None) param_attr=attr)
def Mean(self, node): def Mean(self, node):
input = self.graph.get_node(node.layer.input[0], copy=True) 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.
先完成此消息的编辑!
想要评论请 注册