1 /* Copyright 2016 The TensorFlow Authors. All Rights Reserved.
2 
3 Licensed under the Apache License, Version 2.0 (the "License");
4 you may not use this file except in compliance with the License.
5 You may obtain a copy of the License at
6 
7     http://www.apache.org/licenses/LICENSE-2.0
8 
9 Unless required by applicable law or agreed to in writing, software
10 distributed under the License is distributed on an "AS IS" BASIS,
11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 See the License for the specific language governing permissions and
13 limitations under the License.
14 ==============================================================================*/
15 
16 #ifndef TENSORFLOW_JAVA_SRC_MAIN_NATIVE_OPERATION_JNI_H_
17 #define TENSORFLOW_JAVA_SRC_MAIN_NATIVE_OPERATION_JNI_H_
18 
19 #include <jni.h>
20 
21 #ifdef __cplusplus
22 extern "C" {
23 #endif
24 
25 /*
26  * Class:     org_tensorflow_Operation
27  * Method:    name
28  * Signature: (J)Ljava/lang/String;
29  */
30 JNIEXPORT jstring JNICALL Java_org_tensorflow_Operation_name(JNIEnv *, jclass,
31                                                              jlong);
32 
33 /*
34  * Class:     org_tensorflow_Operation
35  * Method:    type
36  * Signature: (J)Ljava/lang/String;
37  */
38 JNIEXPORT jstring JNICALL Java_org_tensorflow_Operation_type(JNIEnv *, jclass,
39                                                              jlong);
40 
41 /*
42  * Class:     org_tensorflow_Operation
43  * Method:    numOutputs
44  * Signature: (J)I
45  */
46 JNIEXPORT jint JNICALL Java_org_tensorflow_Operation_numOutputs(JNIEnv *,
47                                                                 jclass, jlong);
48 
49 /*
50  * Class:     org_tensorflow_Operation
51  * Method:    outputListLength
52  * Signature: (JLjava/lang/String;)I
53  */
54 JNIEXPORT jint JNICALL Java_org_tensorflow_Operation_outputListLength(JNIEnv *,
55                                                                       jclass,
56                                                                       jlong,
57                                                                       jstring);
58 
59 /*
60  * Class:     org_tensorflow_Operation
61  * Method:    shape
62  * Signature: (JJI)[J
63  */
64 JNIEXPORT jlongArray JNICALL Java_org_tensorflow_Operation_shape(JNIEnv *,
65                                                                  jclass, jlong,
66                                                                  jlong, jint);
67 
68 /*
69  * Class:     org_tensorflow_Operation
70  * Method:    dtype
71  * Signature: (JJI)I
72  */
73 JNIEXPORT jint JNICALL Java_org_tensorflow_Operation_dtype(JNIEnv *, jclass,
74                                                            jlong, jlong, jint);
75 
76 
77 /*
78  * Class:     org_tensorflow_Operation
79  * Method:    inputListLength
80  * Signature: (JLjava/lang/String;)I
81  */
82 JNIEXPORT jint JNICALL Java_org_tensorflow_Operation_inputListLength(JNIEnv *,
83                                                                       jclass,
84                                                                       jlong,
85                                                                       jstring);
86 
87 #ifdef __cplusplus
88 }  // extern "C"
89 #endif  // __cplusplus
90 #endif  // TENSORFLOW_JAVA_SRC_MAIN_NATIVE_OPERATION_JNI_H_
91