(self, rng_states)
Set the RNG state for both the on device and on host RNGs.

        Arguments:
            rng_states (tuple of np.arrays): tuple with 2 elements
                                                1) numpy random number state vector
                                                2) array of uint32 specifying on dev RNG state
Set the RNG state for both the on device and on host RNGs.
Set the RNG state for both the on device and on host RNGs .
def rng_set_state(self, rng_states):
        """
        Set the RNG state for both the on device and on host RNGs.

        Arguments:
            rng_states (tuple of np.arrays): tuple with 2 elements
                                                1) numpy random number state vector
                                                2) array of uint32 specifying on dev RNG state
        """
        # This is probably a CPU-trained model, but loaded on the GPU for
        # predictions
        if len(rng_states) != 2:
            self.rng_reset()
            self.rng.set_state(rng_states)
            return

        self._set_rand_state_dev(state=rng_states[1])
        self.rng.set_state(rng_states[0])
def rng_set_state ( self , rng_states ) : # This is probably a CPU-trained model, but loaded on the GPU for # predictions if len ( rng_states ) != 2 : self . rng_reset ( ) self . rng . set_state ( rng_states ) return self . _set_rand_state_dev ( state = rng_states [ 1 ] ) self . rng . set_state ( rng_states [ 0 ] )