Add indra-ocr MLX conversion (.mlxonnx bundles + meta + card + license)
e41da84 verified 1 day ago
14.1 kB
{"inputs": ["x"], "outputs": ["p"], "nodes": [{"op": "Conv", "inputs": ["x", "onnx::Conv_241", "onnx::Conv_242"], "outputs": ["/m/body/body.0/body.0.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [2, 2]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.0/body.0.0/Conv_output_0"], "outputs": ["/m/body/body.0/body.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.0/body.0.2/HardSwish_output_0", "onnx::Conv_244", "onnx::Conv_245"], "outputs": ["/m/body/body.1/body.1.0/body.1.0.0/Conv_output_0"], "attrs": {"ints": {"group": 16}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [2, 2]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.1/body.1.0/body.1.0.0/Conv_output_0"], "outputs": ["/m/body/body.1/body.1.0/body.1.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.1/body.1.0/body.1.0.2/HardSwish_output_0", "onnx::Conv_247", "onnx::Conv_248"], "outputs": ["/m/body/body.1/body.1.1/body.1.1.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.1/body.1.1/body.1.1.0/Conv_output_0"], "outputs": ["/m/body/body.1/body.1.1/body.1.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.1/body.1.1/body.1.1.2/HardSwish_output_0", "onnx::Conv_250", "onnx::Conv_251"], "outputs": ["/m/body/body.2/body/body.0/body.0.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.2/body/body.0/body.0.0/Conv_output_0"], "outputs": ["/m/body/body.2/body/body.0/body.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.2/body/body.0/body.0.2/HardSwish_output_0", "onnx::Conv_253", "onnx::Conv_254"], "outputs": ["/m/body/body.2/body/body.1/body.1.0/Conv_output_0"], "attrs": {"ints": {"group": 48}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.2/body/body.1/body.1.0/Conv_output_0"], "outputs": ["/m/body/body.2/body/body.1/body.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.2/body/body.1/body.1.2/HardSwish_output_0", "onnx::Conv_256", "onnx::Conv_257"], "outputs": ["/m/body/body.2/body/body.2/body.2.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "Add", "inputs": ["/m/body/body.1/body.1.1/body.1.1.2/HardSwish_output_0", "/m/body/body.2/body/body.2/body.2.0/Conv_output_0"], "outputs": ["/m/body/body.2/Add_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.2/Add_output_0", "onnx::Conv_259", "onnx::Conv_260"], "outputs": ["/m/body/body.3/body.3.0/body.3.0.0/Conv_output_0"], "attrs": {"ints": {"group": 24}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [2, 2]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.3/body.3.0/body.3.0.0/Conv_output_0"], "outputs": ["/m/body/body.3/body.3.0/body.3.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.3/body.3.0/body.3.0.2/HardSwish_output_0", "onnx::Conv_262", "onnx::Conv_263"], "outputs": ["/m/body/body.3/body.3.1/body.3.1.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.3/body.3.1/body.3.1.0/Conv_output_0"], "outputs": ["/m/body/body.3/body.3.1/body.3.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.3/body.3.1/body.3.1.2/HardSwish_output_0", "onnx::Conv_265", "onnx::Conv_266"], "outputs": ["/m/body/body.4/body/body.0/body.0.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.4/body/body.0/body.0.0/Conv_output_0"], "outputs": ["/m/body/body.4/body/body.0/body.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.4/body/body.0/body.0.2/HardSwish_output_0", "onnx::Conv_268", "onnx::Conv_269"], "outputs": ["/m/body/body.4/body/body.1/body.1.0/Conv_output_0"], "attrs": {"ints": {"group": 64}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.4/body/body.1/body.1.0/Conv_output_0"], "outputs": ["/m/body/body.4/body/body.1/body.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.4/body/body.1/body.1.2/HardSwish_output_0", "onnx::Conv_271", "onnx::Conv_272"], "outputs": ["/m/body/body.4/body/body.2/body.2.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "Add", "inputs": ["/m/body/body.3/body.3.1/body.3.1.2/HardSwish_output_0", "/m/body/body.4/body/body.2/body.2.0/Conv_output_0"], "outputs": ["/m/body/body.4/Add_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.4/Add_output_0", "onnx::Conv_274", "onnx::Conv_275"], "outputs": ["/m/body/body.5/body.5.0/body.5.0.0/Conv_output_0"], "attrs": {"ints": {"group": 32}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [2, 2]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.5/body.5.0/body.5.0.0/Conv_output_0"], "outputs": ["/m/body/body.5/body.5.0/body.5.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.5/body.5.0/body.5.0.2/HardSwish_output_0", "onnx::Conv_277", "onnx::Conv_278"], "outputs": ["/m/body/body.5/body.5.1/body.5.1.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.5/body.5.1/body.5.1.0/Conv_output_0"], "outputs": ["/m/body/body.5/body.5.1/body.5.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.5/body.5.1/body.5.1.2/HardSwish_output_0", "onnx::Conv_280", "onnx::Conv_281"], "outputs": ["/m/body/body.6/body/body.0/body.0.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.6/body/body.0/body.0.0/Conv_output_0"], "outputs": ["/m/body/body.6/body/body.0/body.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.6/body/body.0/body.0.2/HardSwish_output_0", "onnx::Conv_283", "onnx::Conv_284"], "outputs": ["/m/body/body.6/body/body.1/body.1.0/Conv_output_0"], "attrs": {"ints": {"group": 96}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.6/body/body.1/body.1.0/Conv_output_0"], "outputs": ["/m/body/body.6/body/body.1/body.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.6/body/body.1/body.1.2/HardSwish_output_0", "onnx::Conv_286", "onnx::Conv_287"], "outputs": ["/m/body/body.6/body/body.2/body.2.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "Add", "inputs": ["/m/body/body.5/body.5.1/body.5.1.2/HardSwish_output_0", "/m/body/body.6/body/body.2/body.2.0/Conv_output_0"], "outputs": ["/m/body/body.6/Add_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.6/Add_output_0", "onnx::Conv_289", "onnx::Conv_290"], "outputs": ["/m/body/body.7/body.7.0/body.7.0.0/Conv_output_0"], "attrs": {"ints": {"group": 48}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [2, 2]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.7/body.7.0/body.7.0.0/Conv_output_0"], "outputs": ["/m/body/body.7/body.7.0/body.7.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.7/body.7.0/body.7.0.2/HardSwish_output_0", "onnx::Conv_292", "onnx::Conv_293"], "outputs": ["/m/body/body.7/body.7.1/body.7.1.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.7/body.7.1/body.7.1.0/Conv_output_0"], "outputs": ["/m/body/body.7/body.7.1/body.7.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.7/body.7.1/body.7.1.2/HardSwish_output_0", "onnx::Conv_295", "onnx::Conv_296"], "outputs": ["/m/body/body.8/body/body.0/body.0.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.8/body/body.0/body.0.0/Conv_output_0"], "outputs": ["/m/body/body.8/body/body.0/body.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.8/body/body.0/body.0.2/HardSwish_output_0", "onnx::Conv_298", "onnx::Conv_299"], "outputs": ["/m/body/body.8/body/body.1/body.1.0/Conv_output_0"], "attrs": {"ints": {"group": 128}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.8/body/body.1/body.1.0/Conv_output_0"], "outputs": ["/m/body/body.8/body/body.1/body.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.8/body/body.1/body.1.2/HardSwish_output_0", "onnx::Conv_301", "onnx::Conv_302"], "outputs": ["/m/body/body.8/body/body.2/body.2.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "Add", "inputs": ["/m/body/body.7/body.7.1/body.7.1.2/HardSwish_output_0", "/m/body/body.8/body/body.2/body.2.0/Conv_output_0"], "outputs": ["/m/body/body.8/Add_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.8/Add_output_0", "onnx::Conv_304", "onnx::Conv_305"], "outputs": ["/m/body/body.9/body.9.0/body.9.0.0/Conv_output_0"], "attrs": {"ints": {"group": 64}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [2, 2]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.9/body.9.0/body.9.0.0/Conv_output_0"], "outputs": ["/m/body/body.9/body.9.0/body.9.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.9/body.9.0/body.9.0.2/HardSwish_output_0", "onnx::Conv_307", "onnx::Conv_308"], "outputs": ["/m/body/body.9/body.9.1/body.9.1.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.9/body.9.1/body.9.1.0/Conv_output_0"], "outputs": ["/m/body/body.9/body.9.1/body.9.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.9/body.9.1/body.9.1.2/HardSwish_output_0", "onnx::Conv_310", "onnx::Conv_311"], "outputs": ["/m/body/body.10/body/body.0/body.0.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.10/body/body.0/body.0.0/Conv_output_0"], "outputs": ["/m/body/body.10/body/body.0/body.0.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.10/body/body.0/body.0.2/HardSwish_output_0", "onnx::Conv_313", "onnx::Conv_314"], "outputs": ["/m/body/body.10/body/body.1/body.1.0/Conv_output_0"], "attrs": {"ints": {"group": 192}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "HardSwish", "inputs": ["/m/body/body.10/body/body.1/body.1.0/Conv_output_0"], "outputs": ["/m/body/body.10/body/body.1/body.1.2/HardSwish_output_0"], "attrs": {}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.10/body/body.1/body.1.2/HardSwish_output_0", "onnx::Conv_316", "onnx::Conv_317"], "outputs": ["/m/body/body.10/body/body.2/body.2.0/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [1, 1], "pads": [0, 0, 0, 0], "strides": [1, 1]}}, "constants": {}}, {"op": "Add", "inputs": ["/m/body/body.9/body.9.1/body.9.1.2/HardSwish_output_0", "/m/body/body.10/body/body.2/body.2.0/Conv_output_0"], "outputs": ["/m/body/body.10/Add_output_0"], "attrs": {}, "constants": {}}, {"op": "ReduceMean", "inputs": ["/m/body/body.10/Add_output_0"], "outputs": ["/m/ReduceMean_output_0"], "attrs": {"ints": {"keepdims": 0}, "intLists": {"axes": [2, 3]}}, "constants": {}}, {"op": "Gemm", "inputs": ["/m/ReduceMean_output_0", "m.head.weight", "m.head.bias"], "outputs": ["/m/head/Gemm_output_0"], "attrs": {"ints": {"transB": 1}, "floats": {"alpha": 1.0, "beta": 1.0}}, "constants": {}}, {"op": "Softmax", "inputs": ["/m/head/Gemm_output_0"], "outputs": ["p"], "attrs": {"ints": {"axis": -1}}, "constants": {}}]}