blob: c3069e07830cfa5cbddb93a3d198eaf75194dcc5 [file] [log] [blame]
ir_version: 3
producer_name: "pytorch"
producer_version: "0.3"
graph {
node {
input: "0"
output: "1"
op_type: "Flatten"
attribute {
name: "axis"
i: 1
type: INT
}
}
name: "torch-jit-export"
input {
name: "0"
type {
tensor_type {
elem_type: FLOAT
shape {
dim {
dim_value: 1
}
}
}
}
}
output {
name: "1"
type {
tensor_type {
elem_type: FLOAT
shape {
dim {
dim_value: 1
}
dim {
dim_value: 1
}
}
}
}
}
}
opset_import {
version: 7
}