Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_cuda.py88 def initialize_cuda_context_rng(): function
H A Dcommon_utils.py2258 from torch.testing._internal.common_cuda import initialize_cuda_context_rng
2259 initialize_cuda_context_rng()