Home
last modified time | relevance | path

Searched refs:op_handle (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc86 JNIEnv* env, jclass clazz, jlong op_handle) { in Java_org_tensorflow_EagerOperationBuilder_delete() argument
87 if (op_handle == 0) return; in Java_org_tensorflow_EagerOperationBuilder_delete()
88 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(op_handle)); in Java_org_tensorflow_EagerOperationBuilder_delete()
92 JNIEnv* env, jclass clazz, jlong op_handle) { in Java_org_tensorflow_EagerOperationBuilder_execute() argument
93 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_execute()
117 JNIEnv* env, jclass clazz, jlong op_handle, jstring device_name) { in Java_org_tensorflow_EagerOperationBuilder_setDevice() argument
118 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_setDevice()
129 JNIEnv* env, jclass clazz, jlong op_handle, jlong input_handle) { in Java_org_tensorflow_EagerOperationBuilder_addInput() argument
130 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_addInput()
141 JNIEnv* env, jclass clazz, jlong op_handle, jlongArray input_handles) { in Java_org_tensorflow_EagerOperationBuilder_addInputList() argument
[all …]
Dgraph_operation_jni.cc82 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_GraphOperation_shape() argument
86 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_GraphOperation_shape()
132 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_GraphOperation_dtype() argument
136 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_GraphOperation_dtype()
Dgraph_operation_builder_jni.cc32 bool resolveOutput(JNIEnv* env, jlong op_handle, jint index, TF_Output* out) { in resolveOutput() argument
33 if (op_handle == 0) { in resolveOutput()
38 out->oper = reinterpret_cast<TF_Operation*>(op_handle); in resolveOutput()
86 JNIEnv* env, jclass clazz, jlong handle, jlong op_handle, jint index) { in Java_org_tensorflow_GraphOperationBuilder_addInput() argument
88 if (!resolveOutput(env, op_handle, index, &out)) return; in Java_org_tensorflow_GraphOperationBuilder_addInput()
124 jlong op_handle) { in Java_org_tensorflow_GraphOperationBuilder_addControlInput() argument
125 if (op_handle == 0) { in Java_org_tensorflow_GraphOperationBuilder_addControlInput()
131 TF_Operation* control = reinterpret_cast<TF_Operation*>(op_handle); in Java_org_tensorflow_GraphOperationBuilder_addControlInput()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc319 void XlaBuilder::IsConstantVisitor(const int64 op_handle, in IsConstantVisitor() argument
322 if (visited->contains(op_handle) || !*is_constant) { in IsConstantVisitor()
327 *(LookUpInstructionByHandle(op_handle).ValueOrDie()); in IsConstantVisitor()
373 visited->insert(op_handle); in IsConstantVisitor()
Dxla_builder.h978 void IsConstantVisitor(const int64 op_handle,