[bug fix] Compute fp16 batch norm in fp32 instead of downcasting the params - #54

[bug fix] Compute fp16 batch norm in fp32 instead of downcasting the params#54

gokulkrishna98 merged 1 commit into

apple:mainapple/coreai-torch:mainfrom

gokulkrishna98:dev/gokul/batchnorm-fp32-computegokulkrishna98/coreai-torch:dev/gokul/batchnorm-fp32-computeCopy head branch name to clipboard