def METHOD_NAME(q, k, v, extra_options):
    #NOTE: In the reference code get_functions takes x (input of the transformer block) as the argument instead of q
    #however from my basic testing it seems that using q instead gives better results
    m, self.u = get_functions(q, ratio, extra_options["original_shape"])
    return m(q), k, v