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