提交 0423b067 编写于 作者: W wjj19950828

Support GFPGAN

上级 8b183f71
...@@ -4501,10 +4501,10 @@ def aten_rand(mapper, graph, node): ...@@ -4501,10 +4501,10 @@ def aten_rand(mapper, graph, node):
layer_inputs = {} layer_inputs = {}
layer_attrs = {} layer_attrs = {}
inputs_name, inputs_node = mapper._get_inputs_name(node) inputs_name, inputs_node = mapper._get_inputs_name(node)
# 获取当前节点输出的list # outputs list
current_outputs = [output_name] current_outputs = [output_name]
current_inputs = [] current_inputs = []
# 处理输入0,即%23,代表end # deal with shape
if inputs_name[0] in mapper.attrs: if inputs_name[0] in mapper.attrs:
layer_attrs["shape"] = mapper.attrs[inputs_name[0]] layer_attrs["shape"] = mapper.attrs[inputs_name[0]]
else: else:
...@@ -4512,7 +4512,7 @@ def aten_rand(mapper, graph, node): ...@@ -4512,7 +4512,7 @@ def aten_rand(mapper, graph, node):
current_outputs, scope_name) current_outputs, scope_name)
layer_inputs["shape"] = inputs_name[0] layer_inputs["shape"] = inputs_name[0]
current_inputs.append(inputs_name[0]) current_inputs.append(inputs_name[0])
# 处理输入1,即%8,代表dtype # deal with dtype
layer_attrs["dtype"] = dtype_dict[mapper.attrs[inputs_name[1]]] layer_attrs["dtype"] = dtype_dict[mapper.attrs[inputs_name[1]]]
graph.add_layer( graph.add_layer(
...@@ -4542,10 +4542,10 @@ def aten_randn(mapper, graph, node): ...@@ -4542,10 +4542,10 @@ def aten_randn(mapper, graph, node):
layer_inputs = {} layer_inputs = {}
layer_attrs = {} layer_attrs = {}
inputs_name, inputs_node = mapper._get_inputs_name(node) inputs_name, inputs_node = mapper._get_inputs_name(node)
# 获取当前节点输出的list # outputs list
current_outputs = [output_name] current_outputs = [output_name]
current_inputs = [] current_inputs = []
# 处理输入0,即%23,代表end # deal with shape
if inputs_name[0] in mapper.attrs: if inputs_name[0] in mapper.attrs:
layer_attrs["shape"] = mapper.attrs[inputs_name[0]] layer_attrs["shape"] = mapper.attrs[inputs_name[0]]
else: else:
...@@ -4553,7 +4553,7 @@ def aten_randn(mapper, graph, node): ...@@ -4553,7 +4553,7 @@ def aten_randn(mapper, graph, node):
current_outputs, scope_name) current_outputs, scope_name)
layer_inputs["shape"] = inputs_name[0] layer_inputs["shape"] = inputs_name[0]
current_inputs.append(inputs_name[0]) current_inputs.append(inputs_name[0])
# 处理输入1,即%8,代表dtype # deal with dtype
layer_attrs["dtype"] = dtype_dict[mapper.attrs[inputs_name[1]]] layer_attrs["dtype"] = dtype_dict[mapper.attrs[inputs_name[1]]]
graph.add_layer( graph.add_layer(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册