Gao, Xiang 5e97f251a8 Enable TF32 support for cuDNN (#40737) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/40737 Reviewed By: mruberry Differential Revision: D22801525 Pulled By: ngimel fbshipit-source-id: ac7f7e728b4b3e01925337e8c9996f26a6433fd2		2020-09-01 15:34:24 -07:00
..
cuda	[reland] Enable TF32 support for cuBLAS (#41498)	2020-07-15 21:00:55 -07:00
cudnn	Enable TF32 support for cuDNN (#40737)	2020-09-01 15:34:24 -07:00
mkl
mkldnn	Add torch.backends.mkldnn.enabled flag (#25459)	2019-09-11 12:09:40 -07:00
openmp	Add torch.backends.openmp.is_available(); fix some cmake messages (#16425)	2019-01-31 16:15:46 -08:00
quantized	Remove fbgemm_is_cpu_supported in favor of torch.backends.quantized.supported_qengines (#26840)	2019-09-27 13:45:15 -07:00
xnnpack	Integrate XNNPACK with custom class for packing weights. (#34047)	2020-03-14 12:51:56 -07:00
__init__.py	Add torch.backends.mkldnn.enabled flag (#25459)	2019-09-11 12:09:40 -07:00