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