depthwise_shapes = [
    (1, 32, 112, 112, 1, 3, 3, 1, 1, 1),
    (1, 16, 112, 112, 6, 3, 3, 2, 1, 1),
    (1, 24, 56, 56, 6, 3, 3, 2, 1, 1),
    (1, 32, 28, 28, 6, 3, 3, 2, 1, 1),
    (1, 64, 14, 14, 6, 3, 3, 1, 1, 1),
    (1, 96, 14, 14, 6, 3, 3, 2, 1, 1),
    (1, 160, 7, 7, 6, 3, 3, 1, 1, 1),
]


depthwise_shapes_b8 = [
    (8, 32, 112, 112, 1, 3, 3, 1, 1, 1),
    (8, 16, 112, 112, 6, 3, 3, 2, 1, 1),
    (8, 24, 56, 56, 6, 3, 3, 2, 1, 1),
    (8, 32, 28, 28, 6, 3, 3, 2, 1, 1),
    (8, 64, 14, 14, 6, 3, 3, 1, 1, 1),
    (8, 96, 14, 14, 6, 3, 3, 2, 1, 1),
    (8, 160, 7, 7, 6, 3, 3, 1, 1, 1),
]


depthwise_shapes_b16 = [
    (16, 32, 112, 112, 1, 3, 3, 1, 1, 1),
    (16, 16, 112, 112, 6, 3, 3, 2, 1, 1),
    (16, 24, 56, 56, 6, 3, 3, 2, 1, 1),
    (16, 32, 28, 28, 6, 3, 3, 2, 1, 1),
    (16, 64, 14, 14, 6, 3, 3, 1, 1, 1),
    (16, 96, 14, 14, 6, 3, 3, 2, 1, 1),
    (16, 160, 7, 7, 6, 3, 3, 1, 1, 1),
]