uniform: change description. -2 (or values below -1 ) means unknown number of arguments or no arguments at all.

Signed-off-by: AbdelRauf <rauf@konduit.ai>
2021-02-23 21:12:16 +01:00 · 2021-02-23 21:12:16 +01:00 · 375efff2e4
parent 426e28640a
commit 375efff2e4
1 changed files with 1 additions and 1 deletions
--- a/libnd4j/include/ops/declarable/generic/random/uniform.cpp
+++ b/libnd4j/include/ops/declarable/generic/random/uniform.cpp
@ -40,7 +40,7 @@ namespace sd {
         * TArgs[0] - min for rng
         * TArgs[1] - max for rng
         */
-        CUSTOM_OP_IMPL(randomuniform, -1, 1, true, 0, -1) {
+        CUSTOM_OP_IMPL(randomuniform, -1, 1, true, 0, -2) {
            // uniform distribution
            auto rng = block.randomGenerator();
            auto dtype = DataType::FLOAT32;