Fix batching in Transformer.
Fix batching in Transformer.

vijaykiran/onyx,ideal-knee/onyx,iperdomo/onyx,KevinGreene/onyx,dignati/onyx,tomasu82/onyx,onyx-platform/onyx,intfrr/onyx,mccraigmccraig/onyx,Deraen/onyx