tvm
tvm copied to clipboard
[Transform] Improvements to LazyTransformParams
-
Handle non-bundled parameters in LazyTransformParams.
-
Check for
"num_input"attribute -
Handle relax.Const in LazyTransformParams
Prior to this commit,
LazyTransformParamswould only output a call to thefset_itemfunction if that element of the output had a correspondingrelax.Binding. Ifrelax.Constappeared in the output, then the call tofset_itemwould be omitted.This commit updates
LazyTransformParamsto check for any non-Varelements of the output tuple.