* polish code details

* remove needless prefix

* revert needless change

* polish grad func generated format