Add --allow-tf32 perf tuning argument that can be used to enable tf32

Defaults to keeping tf32 disabled.  This is because we haven't fully
verified training results with fp32 enabled.