Add keyed_memoize_{method,on_first_arg}