Searched refs:op_handle (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | operation_jni.cc | 88 JNIEnv* env, jclass clazz, jlong graph_handle, jlong op_handle, in Java_org_tensorflow_Operation_shape() argument
92 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_Operation_shape()
141 jlong op_handle, in Java_org_tensorflow_Operation_dtype() argument
145 TF_Operation* op = requireHandle(env, op_handle); in Java_org_tensorflow_Operation_dtype()
|
| D | operation_builder_jni.cc | 33 bool resolveOutput(JNIEnv* env, jlong op_handle, jint index, TF_Output* out) { in resolveOutput() argument
34 if (op_handle == 0) { in resolveOutput()
39 out->oper = reinterpret_cast<TF_Operation*>(op_handle); in resolveOutput()
87 JNIEnv* env, jclass clazz, jlong handle, jlong op_handle, jint index) { in Java_org_tensorflow_OperationBuilder_addInput() argument
89 if (!resolveOutput(env, op_handle, index, &out)) return; in Java_org_tensorflow_OperationBuilder_addInput()
122 JNIEnv* env, jclass clazz, jlong handle, jlong op_handle) { in Java_org_tensorflow_OperationBuilder_addControlInput() argument
123 if (op_handle == 0) { in Java_org_tensorflow_OperationBuilder_addControlInput()
129 TF_Operation* control = reinterpret_cast<TF_Operation*>(op_handle); in Java_org_tensorflow_OperationBuilder_addControlInput()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 197 void XlaBuilder::IsConstantVisitor(const int64 op_handle, in IsConstantVisitor() argument
200 if (visited->contains(op_handle) || !*is_constant) { in IsConstantVisitor()
205 *(LookUpInstructionByHandle(op_handle).ValueOrDie()); in IsConstantVisitor()
252 visited->insert(op_handle); in IsConstantVisitor()
|
| D | xla_builder.h | 637 void IsConstantVisitor(const int64 op_handle,
|