From 375efff2e490fa42dba4d75503242db2c5e828eb Mon Sep 17 00:00:00 2001 From: AbdelRauf Date: Tue, 23 Feb 2021 21:12:16 +0100 Subject: [PATCH] uniform: change description. -2 (or values below -1 ) means unknown number of arguments or no arguments at all. Signed-off-by: AbdelRauf --- libnd4j/include/ops/declarable/generic/random/uniform.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/libnd4j/include/ops/declarable/generic/random/uniform.cpp b/libnd4j/include/ops/declarable/generic/random/uniform.cpp index cb7f146da..7743255d1 100644 --- 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;