def METHOD_NAME(self, op_weights):
    new_seed = tf.random.stateless_uniform((2,), seed=op_weights["seed"], minval=-(2**31), maxval=2**31 - 1)
    new_seed = tf.cast(new_seed, tf.int32)
    op_weights["seed"].assign(new_seed)
    return calc_rb_binary_mask(op_weights["mask"], op_weights["seed"], self.eps)