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

size op fixed 

Signed-off-by: raver119 <raver119@gmail.com>
master
raver119 2020-03-23 17:02:03 +03:00
parent 55ec207eb8
commit 226f0672bc
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
libnd4j/include/ops/declarable/generic/shape/size.cpp
View File

@ -32,6 +32,7 @@ namespace sd {
REQUIRE_TRUE(output->isScalar(), 0, "Size output should be scalar"); REQUIRE_TRUE(output->isScalar(), 0, "Size output should be scalar");
output->p(0, input->lengthOf()); output->p(0, input->lengthOf());
output->syncToDevice();
return Status::OK(); return Status::OK();
} }