Home
last modified time | relevance | path

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

/external/pytorch/torch/_decomp/
Ddecompositions_for_rng.py104 def set_state_from_tensor(self, state): member in PhiloxState
180 def set_state_from_tensor(cls, x): member in PhiloxStateTracker
183 cls.running_state.set_state_from_tensor(x)
/external/pytorch/torch/_functorch/_aot_autograd/
Dtraced_function_transforms.py299 PhiloxStateTracker.set_state_from_tensor(x)