| ir_version: 6 | |
| producer_name: "pytorch" | |
| producer_version: "CURRENT_VERSION" | |
| graph { | |
| node { | |
| input: "0" | |
| output: "1" | |
| output: "2" | |
| name: "Split_0" | |
| op_type: "Split" | |
| attribute { | |
| name: "axis" | |
| i: 0 | |
| type: INT | |
| } | |
| attribute { | |
| name: "split" | |
| ints: 2 | |
| ints: 1 | |
| type: INTS | |
| } | |
| } | |
| name: "torch-jit-export" | |
| input { | |
| name: "0" | |
| type { | |
| tensor_type { | |
| elem_type: 1 | |
| shape { | |
| dim { | |
| dim_value: 3 | |
| } | |
| } | |
| } | |
| } | |
| } | |
| output { | |
| name: "1" | |
| type { | |
| tensor_type { | |
| elem_type: 1 | |
| shape { | |
| dim { | |
| dim_value: 2 | |
| } | |
| } | |
| } | |
| } | |
| } | |
| output { | |
| name: "2" | |
| type { | |
| tensor_type { | |
| elem_type: 1 | |
| shape { | |
| dim { | |
| dim_value: 1 | |
| } | |
| } | |
| } | |
| } | |
| } | |
| } | |
| opset_import { | |
| version: 9 | |
| } |