Home
last modified time | relevance | path

Searched refs:NestedTensorCUDA (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKey.cpp330 {"NestedTensorCUDA", c10::DispatchKey::NestedTensorCUDA}, in parseDispatchKey()
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py196 NestedTensorCUDA = auto() variable in DispatchKey
291 DispatchKey.NestedTensorCUDA,
321 DispatchKey.NestedTensorCUDA,