elvin-n commented on code in PR #14010:
URL: https://github.com/apache/tvm/pull/14010#discussion_r1126114641
##########
python/tvm/driver/tvmc/compiler.py:
##########
@@ -281,6 +279,20 @@ def compile_model(
compilation.
instruments: Optional[Sequence[PassInstrument]]
The list of pass instrument implementations.
+ 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
+ The list of operators to be transformed with desired layout.
+ mixed_precision: bool
+ To enable mixed precision transformation.
+ mixed_precision_ops: list[str], optional
+ The list of operators to be converted to mixed precision.
+ mixed_precision_calculation_type: str
+ The calculation dtype to be used while mixed precision.
Review Comment:
It make sense to describe default values for these new arguments which are
defined in transform.py, `generate_transform_args()`
--
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]
