Home
last modified time | relevance | path

Searched refs:AutogradNestedTensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKeySet.cpp61 {DispatchKey::AutogradNestedTensor, DispatchKey::NestedTensor}) |
140 case DispatchKey::AutogradNestedTensor: in getBackendKeySetFromAutograd()
H A DDispatchKey.cpp132 case DispatchKey::AutogradNestedTensor: in toString()
293 {"AutogradNestedTensor", c10::DispatchKey::AutogradNestedTensor}, in parseDispatchKey()
H A DDispatchKeySet.h653 DispatchKey::AutogradNestedTensor,
771 DispatchKeySet(DispatchKey::AutogradNestedTensor);
H A DDispatchKey.h348 AutogradNestedTensor, enumerator
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DNestedTensorImpl.h197 c10::DispatchKey::AutogradNestedTensor}; in generate_buffer_key_set()
/aosp_15_r20/external/pytorch/torch/nested/_internal/
H A Dnested_tensor.py82 ks = ks.add(DispatchKey.AutogradNestedTensor)
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp709 DEF_ONE(AutogradNestedTensor) in initDispatchBindings()
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py110 AutogradNestedTensor = auto() variable in DispatchKey
/aosp_15_r20/external/pytorch/test/
H A Dtest_python_dispatch.py1324 torch._C._dispatch_keys(x).has(DispatchKey.AutogradNestedTensor)