Searched refs:op_handle (Results 1 – 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/java/src/main/native/ |
D | eager_operation_builder_jni.cc | 86 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 …]
|
D | graph_operation_jni.cc | 82 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()
|
D | graph_operation_builder_jni.cc | 32 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/ |
D | xla_builder.cc | 319 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()
|
D | xla_builder.h | 978 void IsConstantVisitor(const int64 op_handle,
|