Add new args to `BatchSizeFinder` (#21163)
* new args to batch size scaler
* add to tuner
* add to callback
* add testing
* update
* fix tests
* Apply suggestion from @deependujha
* Apply suggestion from @deependujha
* Apply suggestion from @deependujha
* Apply suggestion from @deependujha
* update
* safe default
* update
* add assertion
* fix doc issue
---------
Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com>
Co-authored-by: Deependu <deependujha21@gmail.com>
