| ir_version: 7 |
| producer_name: "pytorch" |
| producer_version: "CURRENT_VERSION" |
| graph { |
| node { |
| input: "onnx::ReduceMean_0" |
| output: "1" |
| name: "ReduceMean_0" |
| op_type: "ReduceMean" |
| attribute { |
| name: "keepdims" |
| i: 0 |
| type: INT |
| } |
| } |
| name: "torch_jit" |
| input { |
| name: "onnx::ReduceMean_0" |
| type { |
| tensor_type { |
| elem_type: 1 |
| 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: 1 |
| shape { |
| } |
| } |
| } |
| } |
| } |
| opset_import { |
| version: 14 |
| } |