🚨🚨🚨 [eomt] make EoMT compatible with pipeline (#39122)
* Make EoMT compatible with pipeline
* Implicit patch offsets
* remove patch offsets from arg
* Modify tests
* Update example
* fix proc testcase
* Add few more args
* add pipeline test suite
* fix
* docstring fixes
* add pipeline test
* changes w.r.t review
* 🙈 MB
* should fix device mismatch
* debug
* Fixes device mismatch
* use decorator
* we can split mlp
* expected values update
---------
Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>