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