未验证 提交 71c69507 编写于 作者: Z Zhanlue Yang 提交者: GitHub

[Eager][YAML] Supported array-type parsing for output tensors (#40058)

上级 ebd0f512
...@@ -213,8 +213,12 @@ def ParseYamlReturns(string): ...@@ -213,8 +213,12 @@ def ParseYamlReturns(string):
returns = [x.strip() for x in string.strip().split(",")] returns = [x.strip() for x in string.strip().split(",")]
for i in range(len(returns)): for i in range(len(returns)):
ret = returns[i] ret_type = returns[i]
returns_list.append(["", ret, i])
assert ret_type in yaml_types_mapping.keys()
ret_type = yaml_types_mapping[ret_type]
returns_list.append(["", ret_type, i])
return returns_list return returns_list
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册