From 226f0672bc0e5d0fa7d0397483d91b623b224ccf Mon Sep 17 00:00:00 2001
From: raver119 <raver119@gmail.com>
Date: Mon, 23 Mar 2020 17:02:03 +0300
Subject: [PATCH] size op fixed

Signed-off-by: raver119 <raver119@gmail.com>
---
 libnd4j/include/ops/declarable/generic/shape/size.cpp | 1 +
 1 file changed, 1 insertion(+)

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();
         }