From 3a902372170c32843e04a1f6f9c172c965c47439 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=D0=9F=D0=B0=D0=B2=D0=B5=D0=BB=20=D0=98=D0=BB=D1=8C=D1=8E?= =?utf8?q?=D1=82=D1=87=D0=B5=D0=BD=D0=BA=D0=BE/AI=20Tools=20Lab=20/SRR/Eng?= =?utf8?q?ineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Thu, 15 Aug 2019 22:08:05 +0300 Subject: [PATCH] [mir_onnx] Fix border type in Average pooling (#6637) * Fix come from debuging InceptionV3 Signed-off-by: Pavel Iliutchenko --- compiler/mir-onnx-importer/Op/AveragePool.cpp | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/compiler/mir-onnx-importer/Op/AveragePool.cpp b/compiler/mir-onnx-importer/Op/AveragePool.cpp index 9aac33e..78df004 100644 --- a/compiler/mir-onnx-importer/Op/AveragePool.cpp +++ b/compiler/mir-onnx-importer/Op/AveragePool.cpp @@ -29,15 +29,13 @@ AveragePoolNodeConverter::convert(const onnx::NodeProto &onnx_node, mir::Graph *graph) const { // TODO Set some asserts - mir::ops::PoolOp::BorderType border_type; - mir::ops::PoolOp::PoolingType pool_type; + mir::ops::PoolOp::BorderType border_type = mir::ops::PoolOp::BorderType::EMPTY; + mir::ops::PoolOp::PoolingType pool_type = mir::ops::PoolOp::PoolingType::AVG; KernelStridesPadding cdata; // Transpose ONNX NCHW to MIR NHWC auto t_input = convertONNXToMIR(graph, inputs[0]); - border_type = mir::ops::PoolOp::BorderType::ZEROFILLED; - pool_type = mir::ops::PoolOp::PoolingType::AVG; getKernelStridesPadding(onnx_node, cdata); auto result = -- 2.7.4