Permalink
Browse files

Fix operator arguments (#4575)

* Add missing argument for activation

* Add type name for nnvm::Tuple<int>
  • Loading branch information...
1 parent 2123ac4 commit 0e9f56a27ed35bbf535384be408e9e9a481bbe06 @lx75249 lx75249 committed with piiswrong Jan 11, 2017
Showing with 2 additions and 0 deletions.
  1. +1 −0 include/mxnet/tensor_blob.h
  2. +1 −0 src/operator/activation.cc
@@ -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 {
@@ -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

0 comments on commit 0e9f56a

Please sign in to comment.