Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_prims/
H A Drng_prims.py120 CUDARngStateHelper.set_torch_state_tensor(seed, offset)
/aosp_15_r20/external/pytorch/torch/_prims_common/
H A D__init__.py1987 def set_torch_state_tensor(seed, offset): member in CUDARngStateHelper