From 3bc04209c6050580f04a5fdb060d58a248f6fa7b Mon Sep 17 00:00:00 2001
From: liuqi <liuqi10@xiaomi.com>
Date: Tue, 19 Sep 2017 17:15:42 +0800
Subject: [PATCH] Add default value to dilation argument of conv2d.

---
 mace/ops/conv_pool_2d_base.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mace/ops/conv_pool_2d_base.h b/mace/ops/conv_pool_2d_base.h
index 96b544d7..9b1838a3 100644
--- a/mace/ops/conv_pool_2d_base.h
+++ b/mace/ops/conv_pool_2d_base.h
@@ -18,7 +18,7 @@ class ConvPool2dOpBase : public Operator<D, T> {
         strides_(OperatorBase::GetRepeatedArgument<int>("strides")),
         padding_(static_cast<Padding>(OperatorBase::GetSingleArgument<int>(
             "padding", static_cast<int>(SAME)))),
-        dilations_(OperatorBase::GetRepeatedArgument<int>("dilations")) {}
+        dilations_(OperatorBase::GetRepeatedArgument<int>("dilations", {1, 1})) {}
 
   void CalOutputSize(const index_t *input_shape,   // NCHW
                      const index_t *filter_shape,  // OIHW
-- 
GitLab