remove template arguments of layernorm
Summary:
remove layernorm templates and make them float since that's the only variant
minor fixes in logging and testing
Test Plan: ran the test
Reviewed By: venkatacrc
Differential Revision: D22527359
fbshipit-source-id: d6eec362a6e88e1c12fddf820ae629ede13fb2b8
