-
Notifications
You must be signed in to change notification settings - Fork 3.4k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Transform] Improvements to LazyTransformParams (#16602)
* [Transform] Improvements to LazyTransformParams * Handle non-bundled parameters in LazyTransformParams. * Check for `"num_input"` attribute * Handle relax.Const in LazyTransformParams Prior to this commit, `LazyTransformParams` would only output a call to the `fset_item` function if that element of the output had a corresponding `relax.Binding`. If `relax.Const` appeared in the output, then the call to `fset_item` would be omitted. This commit updates `LazyTransformParams` to check for any non-`Var` elements of the output tuple. * Update based on review comments
- Loading branch information
1 parent
8fe0164
commit 5308ef1
Showing
2 changed files
with
310 additions
and
20 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.