def _get_perms()

in optimum/quanto/tensor/weights/marlin/permutations.py [0:0]


def _get_perms() -> Tuple[List[int], List[int]]:
    perm = []
    for i in range(8):
        perm.extend([i + 8 * j for j in range(8)])
    perm_single = []
    for i in range(4):
        perm_single.extend([2 * i + j for j in [0, 1, 8, 9, 16, 17, 24, 25]])
    return perm, perm_single