liangfu commented on a change in pull request #4703: [VTA] Support network
which have no unique operator as start/stop name for graph pack.
URL: https://github.com/apache/incubator-tvm/pull/4703#discussion_r367749989
##########
File path: vta/python/vta/top/graphpack.py
##########
@@ -283,15 +289,17 @@ def _recursion(anf, start_found, stop_found):
assert start_found
assert stop_found
return anf
- annotated = _recursion(anf, False, False)
+ annotated = _recursion(anf, False, False, operator_current_idx)
return run_opt_pass(annotated, transform.ToGraphNormalForm())
def graph_pack(expr,
bfactor,
cfactor,
weight_bits,
start_name="nn.max_pool2d",
- stop_name="nn.global_avg_pool2d"):
+ stop_name="nn.global_avg_pool2d",
+ start_name_idx=0,
Review comment:
should be equal to the defaults in `get_subgraph` function.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
With regards,
Apache Git Services
