提交 0afa879b 编写于 作者: W wjj19950828

deal with comments

上级 4e29cc70
...@@ -1242,6 +1242,12 @@ class OpSet9(): ...@@ -1242,6 +1242,12 @@ class OpSet9():
inputs={"input": val_x.name}, inputs={"input": val_x.name},
outputs=[node.name], outputs=[node.name],
**layer_attrs) **layer_attrs)
if val_x.dtype == 'uint8':
self.paddle_graph.add_layer(
'paddle.cast',
inputs={"x": node.name},
outputs=[node.name],
dtype=string('uint8'))
@print_mapping_info @print_mapping_info
def ConstantOfShape(self, node): def ConstantOfShape(self, node):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册