* support scale final state

* fix inplace error

* pass arg directly

* pass arg directly for inplace api

* fix type