[JIT] Include conv3d in the conv-add-relu fusion (#54772)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/54772
conv3d-add-relu fusion does not work on some platforms when TF32 is enabled, so set allow_tf32 to false.
Test Plan:
```
python test/test_jit.py -k test_freeze_conv_relu_fusion
```
Imported from OSS
Reviewed By: bertmaher
Differential Revision: D27435560
fbshipit-source-id: e35e2297dce85acfbe988deea97c3f5e68f1e1c7
