Initial framework wrappers

Signed-off-by: Ryan Nett <[email protected]>

Author: rnett

tensorflow-kotlin-parent/tensorflow-framework-kotlin/pom.xml

@@ -40,6 +40,11 @@
       <artifactId>tensorflow-framework</artifactId>
       <version>${project.version}</version>
     </dependency>
+    <dependency>
+      <groupId>org.tensorflow</groupId>
+      <artifactId>tensorflow-core-kotlin</artifactId>
+      <version>${project.version}</version>
+    </dependency>
     <dependency>
       <groupId>org.junit.jupiter</groupId>
       <artifactId>junit-jupiter-api</artifactId>

tensorflow-kotlin-parent/tensorflow-framework-kotlin/src/main/kotlin/org/tensorflow/framework/activations/Activation.kt

@@ -0,0 +1,32 @@
+/*
+  Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+     http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ =======================================================================
+
+ */
+package org.tensorflow.framework.activations
+
+import org.tensorflow.Operand
+import org.tensorflow.op.kotlin.KotlinOps
+import org.tensorflow.op.kotlin.kotlin
+import org.tensorflow.types.family.TNumber
+
+/**
+ * Create an initializer.
+ * @see org.tensorflow.framework.activations.Activation
+ */
+public inline fun <T : TNumber> Activation(
+    crossinline activation: KotlinOps.(Operand<T>) -> Operand<T>
+): Activation<T> =
+    org.tensorflow.framework.activations.Activation { tf, input -> activation(tf.kotlin, input) }

tensorflow-kotlin-parent/tensorflow-framework-kotlin/src/main/kotlin/org/tensorflow/framework/initializers/Initializer.kt

@@ -0,0 +1,39 @@
+/*
+  Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+
+ Licensed under the Apache License, Version 2.0 (the "License");
+ you may not use this file except in compliance with the License.
+ You may obtain a copy of the License at
+
+     http://www.apache.org/licenses/LICENSE-2.0
+
+ Unless required by applicable law or agreed to in writing, software
+ distributed under the License is distributed on an "AS IS" BASIS,
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ See the License for the specific language governing permissions and
+ limitations under the License.
+ =======================================================================
+
+ */
+package org.tensorflow.framework.initializers
+
+import org.tensorflow.Operand
+import org.tensorflow.op.Ops
+import org.tensorflow.op.kotlin.KotlinOps
+import org.tensorflow.op.kotlin.kotlin
+import org.tensorflow.types.TInt64
+import org.tensorflow.types.family.TType
+
+/**
+ * Create an initializer
+ * @see org.tensorflow.framework.initializers.Initializer
+ */
+public inline fun <T : TType> Initializer(crossinline initializer: KotlinOps.(dims: Operand<TInt64>, dataType: Class<T>) -> Operand<T>): Initializer<T> =
+    org.tensorflow.framework.initializers.Initializer { tf, dims, dataType ->
+      initializer(tf.kotlin, dims, dataType)
+    }
+
+/**
+ * Call an initializer.
+ */
+public inline fun <reified T: TType> Initializer<T>.call(tf: Ops, dims: Operand<TInt64>): Operand<T> = call(tf, dims, T::class.java)!!