diff --git a/cpp-package/scripts/OpWrapperGenerator.py b/cpp-package/scripts/OpWrapperGenerator.py index c09b156758d1..9b08e6b48704 100644 --- a/cpp-package/scripts/OpWrapperGenerator.py +++ b/cpp-package/scripts/OpWrapperGenerator.py @@ -99,7 +99,9 @@ class Arg: 'double or None':'dmlc::optional',\ 'Shape or None':'dmlc::optional',\ 'string':'const std::string&',\ - 'tuple of ':'nnvm::Tuple'} + 'tuple of ':'nnvm::Tuple',\ + 'tuple of <>':'mxnet::cpp::Shape',\ + '':'index_t'} name = '' type = '' description = ''