From 09923a6c22ee36918a33f57acce53e11efdfbb95 Mon Sep 17 00:00:00 2001
From: Roland Grinis <roland.grinis@grinisrit.com>
Date: Mon, 12 Jul 2021 15:48:46 +0100
Subject: [PATCH] update JNI header

---
 .../resources/space_kscience_kmath_noa_JNoa.h | 80 +++++++++++++++++++
 1 file changed, 80 insertions(+)

diff --git a/kmath-noa/src/main/resources/space_kscience_kmath_noa_JNoa.h b/kmath-noa/src/main/resources/space_kscience_kmath_noa_JNoa.h
index adc2ab5ae..12f2685ba 100644
--- a/kmath-noa/src/main/resources/space_kscience_kmath_noa_JNoa.h
+++ b/kmath-noa/src/main/resources/space_kscience_kmath_noa_JNoa.h
@@ -1047,6 +1047,86 @@ JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_autoGradTensor
 JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_autoHessTensor
   (JNIEnv *, jclass, jlong, jlong);
 
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    backwardPass
+ * Signature: (J)V
+ */
+JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_backwardPass
+  (JNIEnv *, jclass, jlong);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    tensorGrad
+ * Signature: (J)J
+ */
+JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_tensorGrad
+  (JNIEnv *, jclass, jlong);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    disposeJitModule
+ * Signature: (J)V
+ */
+JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_disposeJitModule
+  (JNIEnv *, jclass, jlong);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    trainMode
+ * Signature: (JZ)V
+ */
+JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_trainMode
+  (JNIEnv *, jclass, jlong, jboolean);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    loadJitModuleDouble
+ * Signature: (Ljava/lang/String;I)J
+ */
+JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_loadJitModuleDouble
+  (JNIEnv *, jclass, jstring, jint);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    loadJitModuleFloat
+ * Signature: (Ljava/lang/String;I)J
+ */
+JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_loadJitModuleFloat
+  (JNIEnv *, jclass, jstring, jint);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    loadJitModuleLong
+ * Signature: (Ljava/lang/String;I)J
+ */
+JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_loadJitModuleLong
+  (JNIEnv *, jclass, jstring, jint);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    loadJitModuleInt
+ * Signature: (Ljava/lang/String;I)J
+ */
+JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_loadJitModuleInt
+  (JNIEnv *, jclass, jstring, jint);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    forwardPass
+ * Signature: (JJ)J
+ */
+JNIEXPORT jlong JNICALL Java_space_kscience_kmath_noa_JNoa_forwardPass
+  (JNIEnv *, jclass, jlong, jlong);
+
+/*
+ * Class:     space_kscience_kmath_noa_JNoa
+ * Method:    forwardPassAssign
+ * Signature: (JJ)V
+ */
+JNIEXPORT void JNICALL Java_space_kscience_kmath_noa_JNoa_forwardPassAssign
+  (JNIEnv *, jclass, jlong, jlong);
+
 #ifdef __cplusplus
 }
 #endif