Home
last modified time | relevance | path

Searched defs:CustomRNGKeyId (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Drng_extension.cpp60 TORCH_LIBRARY_IMPL(aten, CustomRNGKeyId, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/c10/core/
H A DDispatchKey.h205 CustomRNGKeyId, enumerator
/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dcpu_rng_test.cpp133 TORCH_LIBRARY_IMPL(aten, CustomRNGKeyId, m) { in TORCH_LIBRARY_IMPL() argument
/aosp_15_r20/external/pytorch/torchgen/
H A Dmodel.py91 CustomRNGKeyId = auto() variable in DispatchKey