Lunderberg commented on code in PR #16955:
URL: https://github.com/apache/tvm/pull/16955#discussion_r1594281102
##########
python/tvm/script/parser/relax/entry.py:
##########
@@ -296,13 +298,17 @@ class CallableProxy(StructInfoProxy):
purity : bool
Whether the callable is pure.
+ derive_func: Optional[Union[str, tvm.ir.EnvFunc]]
+ The derivation function for the outputq
Review Comment:
Typo, updated to `output`.
The derivation function is equivalent to `FInferStructInfo` set for relax
operations, but can be applied to an arbitrary function. It can be used in
cases where a PackedFunc may be called, and the output `StructInfo` should be
derived from the input arguments, rather than taken from `sinfo_args`. This
functionality has existed in the C++ API for a while, but it looks like this is
the first time it's been exposed through the Python API, so I fleshed out the
description here.
--
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]
