From e2b1d3432af5b908ecd7e9ce608b651e07412e0b Mon Sep 17 00:00:00 2001 From: channingss Date: Thu, 8 Aug 2019 10:29:21 +0800 Subject: [PATCH] update onnx_op_mapper --- x2paddle/op_mapper/onnx_op_mapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/x2paddle/op_mapper/onnx_op_mapper.py b/x2paddle/op_mapper/onnx_op_mapper.py index a298cca..ebc619c 100644 --- a/x2paddle/op_mapper/onnx_op_mapper.py +++ b/x2paddle/op_mapper/onnx_op_mapper.py @@ -646,7 +646,7 @@ class ONNXOpMapper(OpMapper): val_x = self.graph.get_node(node.layer.input[0], copy=True) val_y = self.graph.get_node(node.layer.output[0], copy=True) input_shape = val_x.out_shapes - output_shape = _val_y.out_shapes + output_shape = val_y.out_shapes assert input_shape is not None or output_shape is not None, 'poolnd not inferred' # N if input_shape: poolnd = len(input_shape) - 2 # NC... -- GitLab