Add ConvertXlaShapeToLazy helper function (#4081)
* Add ConvertXlaShapeToLazy helper function
* Fix build failures during type conversions
* Add unit tests for lazy
* Remove debugging print statements
* Run linter
* Update element_type conversion to an existing util function
* Update minor fixes and add more unit tests
* Update tests
* Run linter again
