| ir_version: 6 |
| producer_name: "pytorch" |
| producer_version: "CURRENT_VERSION" |
| graph { |
| node { |
| input: "input" |
| output: "1" |
| output: "2" |
| output: "3" |
| output: "4" |
| name: "Unique_0" |
| op_type: "Unique" |
| attribute { |
| name: "axis" |
| i: 0 |
| type: INT |
| } |
| attribute { |
| name: "sorted" |
| i: 1 |
| type: INT |
| } |
| } |
| name: "torch-jit-export" |
| input { |
| name: "input" |
| type { |
| tensor_type { |
| elem_type: 1 |
| shape { |
| dim { |
| dim_value: 2 |
| } |
| dim { |
| dim_value: 3 |
| } |
| dim { |
| dim_value: 4 |
| } |
| dim { |
| dim_value: 5 |
| } |
| } |
| } |
| } |
| } |
| output { |
| name: "1" |
| type { |
| tensor_type { |
| elem_type: 1 |
| shape { |
| dim { |
| dim_param: "Unique1_dim_0" |
| } |
| dim { |
| dim_value: 3 |
| } |
| dim { |
| dim_value: 4 |
| } |
| dim { |
| dim_value: 5 |
| } |
| } |
| } |
| } |
| } |
| output { |
| name: "4" |
| type { |
| tensor_type { |
| elem_type: 7 |
| shape { |
| dim { |
| dim_param: "Unique4_dim_0" |
| } |
| } |
| } |
| } |
| } |
| } |
| opset_import { |
| version: 11 |
| } |