diff --git a/libnd4j/include/ops/declarable/generic/shape/size.cpp b/libnd4j/include/ops/declarable/generic/shape/size.cpp index fd76548cb..d31e782c6 100644 --- 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(); }