From 6fd872f0bec48c83636a658f7ccc3c62e4c891ee Mon Sep 17 00:00:00 2001 From: Dmitriy Dolovov <41585329+ddolovov@users.noreply.github.com> Date: Wed, 28 Oct 2020 10:30:43 +0300 Subject: [PATCH] Merge pull request #4470 from JetBrains/master-tensorflow Fix tensorflow sample compilation --- .../src/tensorflowMain/kotlin/HelloTensorflow.kt | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/kotlin-native/samples/tensorflow/src/tensorflowMain/kotlin/HelloTensorflow.kt b/kotlin-native/samples/tensorflow/src/tensorflowMain/kotlin/HelloTensorflow.kt index 04adda9bc89..ec65851f152 100644 --- a/kotlin-native/samples/tensorflow/src/tensorflowMain/kotlin/HelloTensorflow.kt +++ b/kotlin-native/samples/tensorflow/src/tensorflowMain/kotlin/HelloTensorflow.kt @@ -37,11 +37,15 @@ fun scalarTensor(value: Int): Tensor { val data = nativeHeap.allocArray(1) data[0] = value - return TF_NewTensor(TF_INT32, - dims = null, num_dims = 0, - data = data, len = IntVar.size.convert(), - deallocator = staticCFunction { dataToFree, _, _ -> nativeHeap.free(dataToFree!!.reinterpret()) }, - deallocator_arg = null)!! + return TF_NewTensor( + TF_INT32, + /* dims = */ null, + /* num_dims = */ 0, + /* data = */ data, + /* len = */ IntVar.size.convert(), + /* deallocator = */ staticCFunction { dataToFree, _, _ -> nativeHeap.free(dataToFree!!.reinterpret()) }, + /* deallocator_arg = */ null + )!! } val Tensor.scalarIntValue: Int get() {