ir_version: 3 | |
producer_name: "pytorch" | |
producer_version: "0.3" | |
graph { | |
node { | |
input: "0" | |
output: "1" | |
op_type: "Mean" | |
} | |
name: "torch-jit-export" | |
input { | |
name: "0" | |
type { | |
tensor_type { | |
elem_type: FLOAT | |
shape { | |
dim { | |
dim_value: 1 | |
} | |
dim { | |
dim_value: 2 | |
} | |
dim { | |
dim_value: 3 | |
} | |
dim { | |
dim_value: 4 | |
} | |
} | |
} | |
} | |
} | |
output { | |
name: "1" | |
type { | |
tensor_type { | |
elem_type: FLOAT | |
shape { | |
} | |
} | |
} | |
} | |
} | |
opset_import { | |
version: 6 | |
} |