ir_version: 7 | |
producer_name: "pytorch" | |
producer_version: "CURRENT_VERSION" | |
graph { | |
node { | |
input: "onnx::Concat_0" | |
input: "onnx::Concat_1" | |
output: "2" | |
name: "Concat_0" | |
op_type: "Concat" | |
attribute { | |
name: "axis" | |
i: 1 | |
type: INT | |
} | |
} | |
name: "torch_jit" | |
input { | |
name: "onnx::Concat_0" | |
type { | |
tensor_type { | |
elem_type: 1 | |
shape { | |
dim { | |
dim_value: 2 | |
} | |
dim { | |
dim_value: 3 | |
} | |
} | |
} | |
} | |
} | |
input { | |
name: "onnx::Concat_1" | |
type { | |
tensor_type { | |
elem_type: 1 | |
shape { | |
dim { | |
dim_value: 2 | |
} | |
dim { | |
dim_value: 3 | |
} | |
} | |
} | |
} | |
} | |
output { | |
name: "2" | |
type { | |
tensor_type { | |
elem_type: 1 | |
shape { | |
dim { | |
dim_value: 2 | |
} | |
dim { | |
dim_value: 6 | |
} | |
} | |
} | |
} | |
} | |
} | |
opset_import { | |
version: 14 | |
} |