azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #4470 from JetBrains/master-tensorflow

Browse Source 
Fix tensorflow sample compilation
This commit is contained in:
Dmitriy Dolovov
2020-10-28 10:30:43 +03:00
committed by Stanislav Erokhin
parent db3cdddd7a
commit 6fd872f0be
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kotlin-native/samples/tensorflow/src/tensorflowMain/kotlin/HelloTensorflow.kt
+9 -5
View File
@@ -37,11 +37,15 @@ fun scalarTensor(value: Int): Tensor {
val data = nativeHeap.allocArray<IntVar>(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<IntVar>()) },
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<IntVar>()) },
/* deallocator_arg = */ null
)!!
}
val Tensor.scalarIntValue: Int get() {