| {"inputs": ["x", "widths"], "outputs": ["p"], "nodes": [{"op": "AveragePool", "inputs": ["x"], "outputs": ["/m/body/body.0/AveragePool_output_0"], "attrs": {"ints": {"ceil_mode": 0, "count_include_pad": 1}, "intLists": {"kernel_shape": [2, 2], "pads": [0, 0, 0, 0], "strides": [2, 2]}}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.0/AveragePool_output_0", "onnx::Conv_78", "onnx::Conv_79"], "outputs": ["/m/body/body.1/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "Relu", "inputs": ["/m/body/body.1/Conv_output_0"], "outputs": ["/m/body/body.3/Relu_output_0"], "attrs": {}, "constants": {}}, {"op": "MaxPool", "inputs": ["/m/body/body.3/Relu_output_0"], "outputs": ["/m/body/body.4/MaxPool_output_0"], "attrs": {"ints": {"ceil_mode": 0}, "intLists": {"dilations": [1, 1], "kernel_shape": [2, 2], "pads": [0, 0, 0, 0], "strides": [2, 2]}}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.4/MaxPool_output_0", "onnx::Conv_81", "onnx::Conv_82"], "outputs": ["/m/body/body.5/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "Relu", "inputs": ["/m/body/body.5/Conv_output_0"], "outputs": ["/m/body/body.7/Relu_output_0"], "attrs": {}, "constants": {}}, {"op": "MaxPool", "inputs": ["/m/body/body.7/Relu_output_0"], "outputs": ["/m/body/body.8/MaxPool_output_0"], "attrs": {"ints": {"ceil_mode": 0}, "intLists": {"dilations": [1, 1], "kernel_shape": [2, 2], "pads": [0, 0, 0, 0], "strides": [2, 2]}}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.8/MaxPool_output_0", "onnx::Conv_84", "onnx::Conv_85"], "outputs": ["/m/body/body.9/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "Relu", "inputs": ["/m/body/body.9/Conv_output_0"], "outputs": ["/m/body/body.11/Relu_output_0"], "attrs": {}, "constants": {}}, {"op": "MaxPool", "inputs": ["/m/body/body.11/Relu_output_0"], "outputs": ["/m/body/body.12/MaxPool_output_0"], "attrs": {"ints": {"ceil_mode": 0}, "intLists": {"dilations": [1, 1], "kernel_shape": [2, 2], "pads": [0, 0, 0, 0], "strides": [2, 2]}}, "constants": {}}, {"op": "Conv", "inputs": ["/m/body/body.12/MaxPool_output_0", "onnx::Conv_87", "onnx::Conv_88"], "outputs": ["/m/body/body.13/Conv_output_0"], "attrs": {"ints": {"group": 1}, "intLists": {"dilations": [1, 1], "kernel_shape": [3, 3], "pads": [1, 1, 1, 1], "strides": [1, 1]}}, "constants": {}}, {"op": "Relu", "inputs": ["/m/body/body.13/Conv_output_0"], "outputs": ["/m/body/body.15/Relu_output_0"], "attrs": {}, "constants": {}}, {"op": "MaxPool", "inputs": ["/m/body/body.15/Relu_output_0"], "outputs": ["/m/body/body.16/MaxPool_output_0"], "attrs": {"ints": {"ceil_mode": 0}, "intLists": {"dilations": [1, 1], "kernel_shape": [2, 2], "pads": [0, 0, 0, 0], "strides": [2, 2]}}, "constants": {}}, {"op": "ReduceMean", "inputs": ["/m/body/body.16/MaxPool_output_0"], "outputs": ["/m/ReduceMean_output_0"], "attrs": {"ints": {"keepdims": 0}, "intLists": {"axes": [2]}}, "constants": {}}, {"op": "Shape", "inputs": ["/m/ReduceMean_output_0"], "outputs": ["/m/Shape_output_0"], "attrs": {}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_output_0"], "attrs": {}, "constants": {"/m/Constant_output_0": {"kind": "int", "value": 2}}}, {"op": "Gather", "inputs": ["/m/Shape_output_0", "/m/Constant_output_0"], "outputs": ["/m/Gather_output_0"], "attrs": {"ints": {"axis": 0}}, "constants": {}}, {"op": "Cast", "inputs": ["/m/Gather_output_0"], "outputs": ["/m/Cast_output_0"], "attrs": {"ints": {"to": 7}}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_1_output_0"], "attrs": {}, "constants": {"/m/Constant_1_output_0": {"kind": "int", "value": 0}}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_2_output_0"], "attrs": {}, "constants": {"/m/Constant_2_output_0": {"kind": "int", "value": 1}}}, {"op": "Range", "inputs": ["/m/Constant_1_output_0", "/m/Cast_output_0", "/m/Constant_2_output_0"], "outputs": ["/m/Range_output_0"], "attrs": {}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_3_output_0"], "attrs": {}, "constants": {"/m/Constant_3_output_0": {"kind": "intList", "value": [0]}}}, {"op": "Unsqueeze", "inputs": ["/m/Range_output_0", "/m/Constant_3_output_0"], "outputs": ["/m/Unsqueeze_output_0"], "attrs": {}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_4_output_0"], "attrs": {}, "constants": {"/m/Constant_4_output_0": {"kind": "intList", "value": [1]}}}, {"op": "Unsqueeze", "inputs": ["widths", "/m/Constant_4_output_0"], "outputs": ["/m/Unsqueeze_1_output_0"], "attrs": {}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_5_output_0"], "attrs": {}, "constants": {"/m/Constant_5_output_0": {"kind": "int", "value": 32}}}, {"op": "Div", "inputs": ["/m/Unsqueeze_1_output_0", "/m/Constant_5_output_0"], "outputs": ["/m/Div_output_0"], "attrs": {}, "constants": {}}, {"op": "Cast", "inputs": ["/m/Div_output_0"], "outputs": ["/m/Cast_1_output_0"], "attrs": {"ints": {"to": 7}}, "constants": {}}, {"op": "Cast", "inputs": ["/m/Cast_1_output_0"], "outputs": ["/m/Cast_2_output_0"], "attrs": {"ints": {"to": 7}}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_6_output_0"], "attrs": {}, "constants": {"/m/Constant_6_output_0": {"kind": "int", "value": 1}}}, {"op": "Clip", "inputs": ["/m/Cast_2_output_0", "/m/Constant_6_output_0", ""], "outputs": ["/m/Clip_output_0"], "attrs": {}, "constants": {}}, {"op": "Less", "inputs": ["/m/Unsqueeze_output_0", "/m/Clip_output_0"], "outputs": ["/m/Less_output_0"], "attrs": {}, "constants": {}}, {"op": "Cast", "inputs": ["/m/Less_output_0"], "outputs": ["/m/Cast_3_output_0"], "attrs": {"ints": {"to": 1}}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["/m/Constant_7_output_0"], "attrs": {}, "constants": {"/m/Constant_7_output_0": {"kind": "intList", "value": [1]}}}, {"op": "Unsqueeze", "inputs": ["/m/Cast_3_output_0", "/m/Constant_7_output_0"], "outputs": ["/m/Unsqueeze_2_output_0"], "attrs": {}, "constants": {}}, {"op": "Mul", "inputs": ["/m/ReduceMean_output_0", "/m/Unsqueeze_2_output_0"], "outputs": ["/m/Mul_output_0"], "attrs": {}, "constants": {}}, {"op": "Constant", "inputs": [], "outputs": ["onnx::ReduceSum_71"], "attrs": {}, "constants": {"onnx::ReduceSum_71": {"kind": "intList", "value": [-1]}}}, {"op": "ReduceSum", "inputs": ["/m/Mul_output_0", "onnx::ReduceSum_71"], "outputs": ["/m/ReduceSum_output_0"], "attrs": {"ints": {"keepdims": 0}}, "constants": {}}, {"op": "ReduceSum", "inputs": ["/m/Cast_3_output_0", "onnx::ReduceSum_71"], "outputs": ["/m/ReduceSum_1_output_0"], "attrs": {"ints": {"keepdims": 1}}, "constants": {}}, {"op": "Div", "inputs": ["/m/ReduceSum_output_0", "/m/ReduceSum_1_output_0"], "outputs": ["/m/Div_1_output_0"], "attrs": {}, "constants": {}}, {"op": "Gemm", "inputs": ["/m/Div_1_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": {}}]} |