srkreddy1238 commented on code in PR #14010:
URL: https://github.com/apache/tvm/pull/14010#discussion_r1125319146
##########
python/tvm/driver/tvmc/autotuner.py:
##########
@@ -376,12 +375,28 @@ def tune_model(
If using the autoscheduler, write the estimated latency at each step
of tuning to file.
additional_target_options: Optional[Dict[str, Dict[str, Any]]]
Additional target options in a dictionary to combine with initial
Target arguments
+ desired_layout: str, optional
+ Can be one of "NCHW" or "NHWC". When specified, compatible operations
in the graph
+ will have their layout set to this format. Tasks will then be tuned
using this
+ specified layout.
+ desired_layout_ops: list[str], optional
Review Comment:
Handled.
--
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.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
