flops-based/gpt2/9/info.json

{
    "shapes": [
        [
            1,
            13
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            768
        ],
        [
            1,
            13,
            50257
        ]
    ],
    "flops": [
        92112384,
        92112384,
        92112384,
        184224768,
        184224768,
        184224768,
        184224768,
        92162304,
        501765888
    ],
    "times": [
        0.004093170166015625,
        0.0020084381103515625,
        0.001920461654663086,
        0.005430698394775391,
        0.0036170482635498047,
        0.004530429840087891,
        0.00299835205078125,
        0.0017931461334228516,
        0.004969120025634766
    ],
    "params": [
        46471680,
        7087872,
        7087872,
        14175744,
        14175744,
        14175744,
        14175744,
        7089408,
        38597376
    ]
}