blob: be308c5047564b8d349c1426a0e5b8b86d8b04af [file] [log] [blame]
ir_version: 7
producer_name: "pytorch"
producer_version: "CURRENT_VERSION"
graph {
node {
input: "onnx::Gemm_0"
input: "onnx::Gemm_1"
input: "onnx::Gemm_2"
output: "onnx::Gemm_3"
name: "Gemm_0"
op_type: "Gemm"
attribute {
name: "alpha"
f: 1
type: FLOAT
}
attribute {
name: "beta"
f: 1
type: FLOAT
}
}
node {
input: "onnx::Gemm_0"
input: "onnx::Gemm_1"
input: "onnx::Gemm_3"
output: "4"
name: "Gemm_1"
op_type: "Gemm"
attribute {
name: "alpha"
f: 1
type: FLOAT
}
attribute {
name: "beta"
f: 1
type: FLOAT
}
}
name: "torch_jit"
input {
name: "onnx::Gemm_0"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 2
}
dim {
dim_value: 3
}
}
}
}
}
input {
name: "onnx::Gemm_1"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 3
}
dim {
dim_value: 4
}
}
}
}
}
input {
name: "onnx::Gemm_2"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 4
}
}
}
}
}
output {
name: "4"
type {
tensor_type {
elem_type: 1
shape {
dim {
dim_value: 2
}
dim {
dim_value: 4
}
}
}
}
}
}
opset_import {
version: 14
}