pytorch/torch/sparse
2023-07-18 21:07:43 +00:00
..
__init__.py [core][pruning][sparse][feature] SparseSemiStructured tensor subclass (#102135) 2023-06-27 19:21:06 +00:00
_semi_structured_conversions.py Make conversions from/to sparse semi-structured always @torch.compile-d (#105272) 2023-07-18 04:51:28 +00:00
_triton_ops.py SDPA: frontend for BSR masks (#104042) 2023-07-13 18:01:21 +00:00
semi_structured.py [BE] Enable ruff's UP rules and autoformat optim/ (#105426) 2023-07-18 21:07:43 +00:00