Fix operator arguments (#4575)

* Add missing argument for activation * Add type name for nnvm::Tuple<int>
apache · Jan 11, 2017 · 0e9f56a · 0e9f56a
1 parent 2123ac4
commit 0e9f56a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/include/mxnet/tensor_blob.h b/include/mxnet/tensor_blob.h
@@ -283,6 +283,7 @@ class TBlob {
 namespace dmlc {
 // Add a few patches to support TShape in dmlc/parameter.
 DMLC_DECLARE_TYPE_NAME(mxnet::TShape, "Shape(tuple)");
+DMLC_DECLARE_TYPE_NAME(nnvm::Tuple<int>, "Shape(tuple)");
 
 namespace parameter {
 

diff --git a/src/operator/activation.cc b/src/operator/activation.cc
@@ -77,6 +77,7 @@ scalar of the input tensor):
 
 See `LeakyReLU` for other activations with parameters.
 )")
+.add_argument("data", "Symbol", "Input data to activation function.")
 .add_arguments(ActivationParam::__FIELDS__());
 
 }  // namespace op