* REF_CODE_REFACTOR: logistic
Refactoring the reference code for logistic operator.

BUG=refactoring existing code.

* Fix formatting
Co-authored-by: NAdvait Jain <advaitjain@users.noreply.github.com>
Co-authored-by: NAdvait Jain <advaitjain@google.com>