pytorch/torch/backends
2024-10-24 17:27:06 +00:00
..
_coreml
_nnapi Remove unused Python variables in torch/[b-z]* (#136963) 2024-10-19 16:45:22 +00:00
cpu Extending the Pytorch vec backend for SVE (ARM) (#119571) 2024-09-18 18:59:10 +00:00
cuda [ROCm] CK-based GEMM (#131004) 2024-10-20 02:57:43 +00:00
cudnn
cusparselt Revert "[sparse] add search for optimal alg_id to torch.compile (#137427)" 2024-10-24 17:27:06 +00:00
mha
mkl
mkldnn
mps
nnpack
openmp
opt_einsum Clarify opt-einsum usage, fix #127109 (#137596) 2024-10-09 20:31:24 +00:00
quantized
xeon
xnnpack
__init__.py