From 2ab395c8c54a814cff8973858d251f47e9dd88b9 Mon Sep 17 00:00:00 2001
From: Haibin Lin <linhaibin.eric@gmail.com>
Date: Mon, 5 Nov 2018 16:23:54 -0800
Subject: [PATCH] Add missing documentations for getnnz

---
 src/operator/contrib/nnz.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/operator/contrib/nnz.cc b/src/operator/contrib/nnz.cc
index 956b22264667..a94f52fdaa5e 100644
--- a/src/operator/contrib/nnz.cc
+++ b/src/operator/contrib/nnz.cc
@@ -182,7 +182,8 @@ This operator only supports CSR matrix on CPU.
 .set_attr<nnvm::FInferType>("FInferType", NNZType)
 .set_attr<FInferStorageType>("FInferStorageType", NNZStorageType)
 .set_attr<FComputeEx>("FComputeEx<cpu>", NNZComputeEx<cpu>)
-.add_argument("data", "NDArray-or-Symbol", "Input");
+.add_argument("data", "NDArray-or-Symbol", "Input")
+.add_arguments(NNZParam::__FIELDS__());
 
 }  // namespace op
 }  // namespace mxnet