@def_function.function
def f(x, y):
    return math_ops.matmul(x, array_ops.reshape(array_ops.transpose(y), [384, 1536]))
@def_function.function
def f(x, y):
    return math_ops.matmul(x, array_ops.reshape(array_ops.transpose(y), [384, 1536]))