提交 d2674430 编写于 作者: S SunAhong1993

fix tolist

上级 75187123
...@@ -921,7 +921,7 @@ class TFOpMapper(OpMapper): ...@@ -921,7 +921,7 @@ class TFOpMapper(OpMapper):
# outputs=[reshape_name], # outputs=[reshape_name],
# shape=shape) # shape=shape)
# inputs['offsets'] = reshape_name # inputs['offsets'] = reshape_name
begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).to_list() begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).tolist()
attrs['offsets'] = begin attrs['offsets'] = begin
if size.layer_type == "Const": if size.layer_type == "Const":
size = size.value.tolist() size = size.value.tolist()
......
...@@ -897,7 +897,7 @@ class TFOpMapper(OpMapper): ...@@ -897,7 +897,7 @@ class TFOpMapper(OpMapper):
# outputs=[reshape_name], # outputs=[reshape_name],
# shape=shape) # shape=shape)
# inputs['offsets'] = reshape_name # inputs['offsets'] = reshape_name
begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).to_list() begin = self.decoder.infer_tensor(begin, use_diff_inputs=False).tolist()
attrs['offsets'] = begin attrs['offsets'] = begin
if size.layer_type == "Const": if size.layer_type == "Const":
size = size.value.tolist() size = size.value.tolist()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册