baberu-ocr-webgpu / reports /encoder-report.json
ameraino11's picture
Add complete 121 MB and 242 MB WebGPU variants, port source, and benchmarks
520f98d verified
Raw
History Blame Contribute Delete
2.33 kB
{
"fp32": {
"graph": {
"bytes": 349172734,
"sha256": "b1ba05ca0848d8bf79d8a0e29e7b69a84245a48c0000add1535036ed56ade377",
"opsets": {
"ai.onnx": 17
},
"operators": {
"Add": 112,
"Cast": 4,
"Concat": 53,
"Constant": 301,
"ConstantOfShape": 1,
"Conv": 1,
"Div": 13,
"Equal": 1,
"Erf": 13,
"Expand": 1,
"Gather": 36,
"LayerNormalization": 25,
"MatMul": 98,
"Mul": 75,
"Pow": 1,
"Reshape": 51,
"Resize": 1,
"Shape": 38,
"Slice": 3,
"Softmax": 12,
"Transpose": 51,
"Unsqueeze": 62,
"Where": 1
},
"inputs": {
"pixel_values": [
1,
3,
224,
224
]
},
"outputs": {
"vision_embeds": [
"Addvision_embeds_dim_0",
"Addvision_embeds_dim_1",
512
]
}
},
"parity": {
"max_abs": 0.00010710954666137695,
"mean_abs": 1.4819087823525479e-07,
"cosine_similarity": 0.9999999999602387
}
},
"official_fp16": {
"graph": {
"bytes": 172917304,
"sha256": "4ce333804846b23f1983376efe363c164d4624ecc6bd16f99a15a859d8423117",
"opsets": {
"ai.onnx": 17,
"com.microsoft.experimental": 1,
"ai.onnx.ml": 5,
"ai.onnx.training": 1,
"com.microsoft": 1,
"ai.onnx.preview.training": 1,
"com.microsoft.nchwc": 1,
"org.pytorch.aten": 1
},
"operators": {
"Add": 1,
"Cast": 2,
"Concat": 1,
"Conv": 1,
"Gelu": 13,
"Gemm": 74,
"LayerNormalization": 1,
"MatMul": 24,
"Mul": 48,
"Reshape": 149,
"SkipLayerNormalization": 24,
"Slice": 1,
"Softmax": 12,
"Transpose": 49
},
"inputs": {
"pixel_values": [
1,
3,
224,
224
]
},
"outputs": {
"vision_embeds": [
1,
256,
512
]
}
},
"parity_to_pytorch_fp32": {
"max_abs": 0.058362603187561035,
"mean_abs": 6.321475666482002e-05,
"cosine_similarity": 0.9999902873467554
}
}
}