Changes to add bfloat16 support for image rotation
Hi @jojivk73 Can you please check @cantonios's comments and keep us posted ? Thank you! @gbaned @cantonios. Sorry for the delay. I updated the name for testcase.
Thanks
These are changes to add bfloat16 for image rotation. With mixed bfloat16 precision, models that use image transformation would perform much better with image rotation in bfloat16.