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_BUILDER_JNI_H_ 17 #define TENSORFLOW_JAVA_SRC_MAIN_NATIVE_OPERATION_BUILDER_JNI_H_ 18 19 #include <jni.h> 20 21 #ifdef __cplusplus 22 extern "C" { 23 #endif 24 25 /* 26 * Class: org_tensorflow_OperationBuilder 27 * Method: allocate 28 * Signature: (JLjava/lang/String;Ljava/lang/String;)J 29 */ 30 JNIEXPORT jlong JNICALL Java_org_tensorflow_OperationBuilder_allocate( 31 JNIEnv *, jclass, jlong, jstring, jstring); 32 33 /* 34 * Class: org_tensorflow_OperationBuilder 35 * Method: finish 36 * Signature: (J)J 37 */ 38 JNIEXPORT jlong JNICALL Java_org_tensorflow_OperationBuilder_finish(JNIEnv *, 39 jclass, 40 jlong); 41 42 /* 43 * Class: org_tensorflow_OperationBuilder 44 * Method: addInput 45 * Signature: (JJI)V 46 */ 47 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_addInput( 48 JNIEnv *, jclass, jlong, jlong, jint); 49 50 /* 51 * Class: org_tensorflow_OperationBuilder 52 * Method: addInputList 53 * Signature: (J[J[I)V 54 */ 55 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_addInputList( 56 JNIEnv *, jclass, jlong, jlongArray, jintArray); 57 58 /* 59 * Class: org_tensorflow_OperationBuilder 60 * Method: addControlInput 61 * Signature: (JJ)V 62 */ 63 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_addControlInput( 64 JNIEnv *, jclass, jlong, jlong); 65 66 /* 67 * Class: org_tensorflow_OperationBuilder 68 * Method: setDevice 69 * Signature: (JLjava/lang/String;)V 70 */ 71 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setDevice(JNIEnv *, 72 jclass, 73 jlong, 74 jstring); 75 76 /* 77 * Class: org_tensorflow_OperationBuilder 78 * Method: setAttrString 79 * Signature: (JLjava/lang/String;[B)V 80 */ 81 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrString( 82 JNIEnv *, jclass, jlong, jstring, jbyteArray); 83 84 /* 85 * Class: org_tensorflow_OperationBuilder 86 * Method: setAttrInt 87 * Signature: (JLjava/lang/String;J)V 88 */ 89 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrInt( 90 JNIEnv *, jclass, jlong, jstring, jlong); 91 92 /* 93 * Class: org_tensorflow_OperationBuilder 94 * Method: setAttrIntList 95 * Signature: (JLjava/lang/String;[J)V 96 */ 97 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrIntList( 98 JNIEnv *, jclass, jlong, jstring, jlongArray); 99 100 /* 101 * Class: org_tensorflow_OperationBuilder 102 * Method: setAttrFloat 103 * Signature: (JLjava/lang/String;F)V 104 */ 105 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrFloat( 106 JNIEnv *, jclass, jlong, jstring, jfloat); 107 108 /* 109 * Class: org_tensorflow_OperationBuilder 110 * Method: setAttrFloatList 111 * Signature: (JLjava/lang/String;[F)V 112 */ 113 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrFloatList( 114 JNIEnv *, jclass, jlong, jstring, jfloatArray); 115 116 /* 117 * Class: org_tensorflow_OperationBuilder 118 * Method: setAttrBool 119 * Signature: (JLjava/lang/String;Z)V 120 */ 121 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrBool( 122 JNIEnv *, jclass, jlong, jstring, jboolean); 123 124 /* 125 * Class: org_tensorflow_OperationBuilder 126 * Method: setAttrBoolList 127 * Signature: (JLjava/lang/String;[Z)V 128 */ 129 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrBoolList( 130 JNIEnv *, jclass, jlong, jstring, jbooleanArray); 131 132 /* 133 * Class: org_tensorflow_OperationBuilder 134 * Method: setAttrType 135 * Signature: (JLjava/lang/String;I)V 136 */ 137 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrType( 138 JNIEnv *, jclass, jlong, jstring, jint); 139 140 /* 141 * Class: org_tensorflow_OperationBuilder 142 * Method: setAttrTypeList 143 * Signature: (JLjava/lang/String;[I)V 144 */ 145 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrTypeList( 146 JNIEnv *, jclass, jlong, jstring, jintArray); 147 148 /* 149 * Class: org_tensorflow_OperationBuilder 150 * Method: setAttrTensor 151 * Signature: (JLjava/lang/String;J)V 152 */ 153 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrTensor( 154 JNIEnv *, jclass, jlong, jstring, jlong); 155 156 /* 157 * Class: org_tensorflow_OperationBuilder 158 * Method: setAttrTensorList 159 * Signature: (JLjava/lang/String;[J)V 160 */ 161 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrTensorList( 162 JNIEnv *, jclass, jlong, jstring, jlongArray); 163 164 /* 165 * Class: org_tensorflow_OperationBuilder 166 * Method: setAttrShape 167 * Signature: (JLjava/lang/String;[JI)V 168 */ 169 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrShape( 170 JNIEnv *, jclass, jlong, jstring, jlongArray, jint); 171 172 /* 173 * Class: org_tensorflow_OperationBuilder 174 * Method: setAttrShapeList 175 * Signature: (JLjava/lang/String;[J[I)V 176 */ 177 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrShapeList( 178 JNIEnv *, jclass, jlong, jstring, jlongArray, jintArray); 179 180 /* 181 * Class: org_tensorflow_OperationBuilder 182 * Method: setAttrStringList 183 * Signature: (JLjava/lang/String;[L)V 184 */ 185 JNIEXPORT void JNICALL Java_org_tensorflow_OperationBuilder_setAttrStringList( 186 JNIEnv *, jclass, jlong, jstring, jobjectArray); 187 188 #ifdef __cplusplus 189 } // extern "C" 190 #endif // __cplusplus 191 #endif // TENSORFLOW_JAVA_SRC_MAIN_NATIVE_OPERATION_BUILDER_JNI_H_ 192