From fa5a47c1fe157f545ad156ffbe301901e8abf206 Mon Sep 17 00:00:00 2001
From: AnastasiaTuchina <anastasia.tuchina@jetbrains.com>
Date: Mon, 17 Jul 2023 15:11:50 +0200
Subject: [PATCH] KI-40 [tfjs] Remove casting of output data in Size operator

---
 .../jsMain/kotlin/io.kinference.tfjs/operators/tensor/Size.kt   | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference/inference-tfjs/src/jsMain/kotlin/io.kinference.tfjs/operators/tensor/Size.kt b/inference/inference-tfjs/src/jsMain/kotlin/io.kinference.tfjs/operators/tensor/Size.kt
index 5198463aa..744557d46 100644
--- a/inference/inference-tfjs/src/jsMain/kotlin/io.kinference.tfjs/operators/tensor/Size.kt
+++ b/inference/inference-tfjs/src/jsMain/kotlin/io.kinference.tfjs/operators/tensor/Size.kt
@@ -52,7 +52,7 @@ class SizeVer1(
 
     override suspend fun <D : ONNXData<*, *>> apply(contexts: Contexts<D>, inputs: List<TFJSTensor?>): List<TFJSTensor?> {
         val inputSize = inputs[0]!!.data.linearSize
-        val dataSize = NDArrayTFJS.intScalar(inputSize.toLong())
+        val dataSize = NDArrayTFJS.intScalar(inputSize)
         return listOf(dataSize.asTensor("size"))
     }
 }