code: def _raise_key(self, key):
      pk = [self._config.PHILOX_KEY_A, self._config.PHILOX_KEY_B]
      return key + np.array(pk, dtype=self._dtype)
  
imports: import numpy
  import pytest
  import scipy.stats
  import torch
  import triton
  import triton.language
label_desc: Use image node_id 6 for calling the CustomPhilox4x obj's underlying member method code with example usage: obj._raise_key(key) and returns: unknown