Add assertion to make sure init op is always fp16 compatible in fp16 training
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/18498
Reviewed By: kennyhorror
Differential Revision: D14626755
fbshipit-source-id: d8a0b3c02920ab3835911a21bf05e8956853fcd7