Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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>
master
AbdelRauf 2021-02-23 21:12:16 +01:00
parent 426e28640a
commit 375efff2e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
libnd4j/include/ops/declarable/generic/random/uniform.cpp
View File

@ -40,7 +40,7 @@ namespace sd {
* TArgs[0] - min for rng * TArgs[0] - min for rng
* TArgs[1] - max 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 // uniform distribution
auto rng = block.randomGenerator(); auto rng = block.randomGenerator();
auto dtype = DataType::FLOAT32; auto dtype = DataType::FLOAT32;