frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add NestedTensor python subclass (#108314)

Browse Source 
Description coming soon

Pull Request resolved: https://github.com/pytorch/pytorch/pull/108314
Approved by: https://github.com/jbschlosser
ghstack dependencies: #108808
This commit is contained in:
soulitzer
2023-09-08 13:24:21 -04:00
committed by PyTorch MergeBot
parent 4a4a2fc1a5
commit 2bcff92540
9 changed files with 465 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torchgen/model.py
View File

@ -84,6 +84,7 @@ class DispatchKey(Enum):
Sparse = auto()
SparseCsrCPU = auto()
SparseCsrCUDA = auto()
NestedTensor = auto()
Python = auto()
FuncTorchDynamicLayerBackMode = auto()