Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Fix operator arguments #4575

Merged
merged 2 commits into from Jan 11, 2017
Merged

Fix operator arguments #4575

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
0121927
Add missing argument for activation
Jan 7, 2017
eb47bb3
Add type name for nnvm::Tuple<int>
Jan 8, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Jump to
Jump to file
Failed to load files.
Diff view
Diff view
1 change: 1 addition & 0 deletions include/mxnet/tensor_blob.h
View file
Open in desktop
Expand Up @@ -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 {

Expand Down
1 change: 1 addition & 0 deletions src/operator/activation.cc
View file
Open in desktop
Expand Up @@ -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
Expand Down