size op fixed

Signed-off-by: raver119 <raver119@gmail.com>
2020-03-23 17:02:03 +03:00 · 2020-03-23 17:02:03 +03:00 · 226f0672bc
commit 226f0672bc
parent 55ec207eb8
1 changed files with 1 additions and 0 deletions
--- a/libnd4j/include/ops/declarable/generic/shape/size.cpp
+++ b/libnd4j/include/ops/declarable/generic/shape/size.cpp
@ -32,6 +32,7 @@ namespace sd {
            REQUIRE_TRUE(output->isScalar(), 0, "Size output should be scalar");

            output->p(0, input->lengthOf());
+            output->syncToDevice();

            return Status::OK();
        }