ir_version: 8 producer_name: "pytorch" producer_version: "CURRENT_VERSION" graph { node { output: "onnx::Split_1" name: "Constant_1" op_type: "Constant" attribute { name: "value" t { dims: 3 data_type: 7 raw_data: "\002\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000\002\000\000\000\000\000\000\000" } type: TENSOR } } node { input: "tensor" input: "onnx::Split_1" output: "2" output: "3" output: "4" name: "Split_2" op_type: "Split" attribute { name: "axis" i: 1 type: INT } } name: "main_graph" input { name: "tensor" type { tensor_type { elem_type: 1 shape { dim { dim_value: 2 } dim { dim_value: 6 } } } } } output { name: "2" type { tensor_type { elem_type: 1 shape { dim { dim_value: 2 } dim { dim_value: 2 } } } } } output { name: "3" type { tensor_type { elem_type: 1 shape { dim { dim_value: 2 } dim { dim_value: 2 } } } } } output { name: "4" type { tensor_type { elem_type: 1 shape { dim { dim_value: 2 } dim { dim_value: 2 } } } } } } opset_import { version: 17 }