From aba4087091cac8ee108c51020ea8628f51d8dfde Mon Sep 17 00:00:00 2001
From: lihongkang <[lihongkang1@huawei.com]>
Date: Tue, 29 Dec 2020 11:32:30 +0800
Subject: [PATCH] fix bugs

---
 mindspore/nn/layer/quant.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/mindspore/nn/layer/quant.py b/mindspore/nn/layer/quant.py
index 491ff3f9d8..d0dcd153c5 100644
--- a/mindspore/nn/layer/quant.py
+++ b/mindspore/nn/layer/quant.py
@@ -348,6 +348,9 @@ class Conv2dBnFoldQuantOneConv(Cell):
         padding (int): Implicit paddings on both sides of the input. Default: 0.
         eps (float): Parameters for BatchNormal. Default: 1e-5.
         momentum (float): Parameters for BatchNormal op. Default: 0.997.
+        dilation (int): Specifies the dilation rate to use for dilated convolution. Default: 1.
+        group (int): Splits filter into groups, `in_ channels` and `out_channels` must be
+            divisible by the number of groups. Default: 1.
         has_bias (bool): Specifies whether the layer uses a bias vector. Default: False.
         weight_init (Union[Tensor, str, Initializer, numbers.Number]): Initializer for the
             convolution kernel. Default: 'normal'.