feat: allow mixed precision policy as dtype (#3751)
* feat: allow mixed precision as dtype
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
* feat: allow mixed precision as dtype
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
* feat: allow mixed precision as dtype
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
* test: extend test for MP as str dtype
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
* Fix: style
---------
Signed-off-by: Mehant Kammakomati <mehant.kammakomati2@ibm.com>
Co-authored-by: S1ro1 <matej.sirovatka@gmail.com>
