Skip to content

Commit 23037a9

Browse files
committed
add has_multi_outs
1 parent caeb961 commit 23037a9

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/mrt/sym_utils.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -471,6 +471,9 @@ def get_entry_id(sym):
471471
oindex = json.loads(graph.json())['heads'][0][1]
472472
return oindex
473473

474+
def has_multi_outs(sym):
475+
return sym.attr('op_name') in MULTIPYE_OUTS_NODE
476+
474477
def get_node(sym, graph):
475478
""" Get the symbol from the provided graph which has the same name as the given symbol.
476479

0 commit comments

Comments
 (0)