Skip to content

T
tic
Unverified Commit e54a9a97 by Tianqi Chen Committed by GitHub
Options

[TUTORIAL] Fix vta tutorial after relay function refactor (#5095)

parent b64a843a
Showing with 5 additions and 5 deletions
vta/python/vta/top/graphpack.py
...@@ -368,11 +368,11 @@ def get_subgraph(expr, start_name, stop_name, start_name_idx, stop_name_idx, cou ...@@ -368,11 +368,11 @@ def get_subgraph(expr, start_name, stop_name, start_name_idx, stop_name_idx, cou
def _recursion(anf, start_found, stop_found, operator_current_idx): def _recursion(anf, start_found, stop_found, operator_current_idx):
""" Helper to obtain the subgraph. """ Helper to obtain the subgraph.
""" """
if isinstance(anf, relay.expr.Function): if isinstance(anf, relay.Function):
return relay.expr.Function(anf.params, return relay.Function(anf.params,
_recursion(anf.body, start_found, stop_found, _recursion(anf.body, start_found, stop_found,
operator_current_idx), operator_current_idx),
anf.ret_type, anf.type_params, anf.attrs) anf.ret_type, anf.type_params, anf.attrs)
if isinstance(anf, relay.expr.Let): if isinstance(anf, relay.expr.Let):
value = anf.value value = anf.value
if isinstance(value, relay.expr.Call): if isinstance(value, relay.expr.Call):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment