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_r369882107
##########
File path: vta/python/vta/top/graphpack.py
##########
@@ -309,18 +329,30 @@ def graph_pack(expr,
The bit-width of the weights.
start_name: str, optional
- Start packing from certain known node.
+ Start packing from certain known node when start_name_idx is None.
stop_name: str, optional
- Stop packing from certain known node.
+ Stop packing from certain known node when stop_name_idx is None.
+
+ start_name_idx: str, optional
+ When start_name_idx not None, start packing only when node name equal
start_name
+ and node idx equal start_name_idx.
+
+ stop_name_idx: str, optional
+ When stop_name_idx not None, stop packing only when node name equal
stop_name
+ and node index equal stop_name_idx.
+
+ count_meta:boolean, optional
+ start_name_idx and stop_name_idx count meta or not.
Review comment:
Thanks for the clear explanation. I think it would be good to have the
explanation in this line of code.
----------------------------------------------------------------
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
