Make test for cast to/from int/float16 and make sure to have a cast in the inner graph instead of identify. (it was still working with identity, but having an explicit cast is better)