Searched refs:set_state_from_tensor (Results 1 – 2 of 2) sorted by relevance
104 def set_state_from_tensor(self, state): member in PhiloxState180 def set_state_from_tensor(cls, x): member in PhiloxStateTracker183 cls.running_state.set_state_from_tensor(x)
299 PhiloxStateTracker.set_state_from_tensor(x)