Closed analog-cbarber closed 6 years ago
It appears that adding input argument to registry declaration fixes the problem:
MXNET_REGISTER_OP_PROPERTY(QActivation, QActivationProp)
.describe(R"(Quantized activation function.
The following quantized/binarized activation are supported (operations are applied elementwisely to each
scalar of the input tensor):
- `1 bit`: using deteministic sign() function to generate binary activation
- `2-32 bit`: using quantization function
)")
.add_argument("data", "NDArray-or-Symbol", "Input data.")
.add_arguments(QActivationParam::__FIELDS__());
thanks for reporting the issue, has been fixed.
Description
Support new Q* operators in mxnet.ndarray module.
Details
The new operators are visible in the
mxnet.ndarray
module but when you try to invoke the input argument is not handled correctly. For example:Environment info (Required)
Built on a macbook pro with USE_CUDA and USE_OPENMP turned off. Python 2.7
Built from source as of commit e4fc9106e9ada87386407375e6297833e0cd76ad.
Note that I had to change the library type from MODULE to SHARED in the root CMakeList.txt file for the cmake build to succeed.