Home
last modified time | relevance | path

Searched defs:op_handle (Results 1 – 13 of 13) 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()
92 JNIEnv* env, jclass clazz, jlong 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()
129 JNIEnv* env, jclass clazz, jlong op_handle, jlong input_handle) { in Java_org_tensorflow_EagerOperationBuilder_addInput()
141 JNIEnv* env, jclass clazz, jlong op_handle, jlongArray input_handles) { in Java_org_tensorflow_EagerOperationBuilder_addInputList()
163 JNIEnv* env, jclass clazz, jlong op_handle, jstring attr_name, in Java_org_tensorflow_EagerOperationBuilder_setAttrString()
178 JNIEnv* env, jclass object, jlong op_handle, jstring attr_name, in Java_org_tensorflow_EagerOperationBuilder_setAttrStringList()
273 JNIEnv* env, jclass clazz, jlong op_handle, jstring attr_name, in Java_org_tensorflow_EagerOperationBuilder_setAttrShape()
299 JNIEnv* env, jclass clazz, jlong op_handle, jstring attr_name, in Java_org_tensorflow_EagerOperationBuilder_setAttrShapeList()
Dgraph_operation_jni.cc82 JNIEnv* env, jclass clazz, jlong graph_handle, jlong 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()
Dgraph_operation_builder_jni.cc32 bool resolveOutput(JNIEnv* env, jlong op_handle, jint index, TF_Output* out) { in resolveOutput()
86 JNIEnv* env, jclass clazz, jlong handle, jlong op_handle, jint index) { in Java_org_tensorflow_GraphOperationBuilder_addInput()
124 jlong op_handle) { in Java_org_tensorflow_GraphOperationBuilder_addControlInput()
/external/pytorch/torch/csrc/jit/mobile/compatibility/
Druntime_compatibility.cpp54 const auto op_handle = c10::Dispatcher::singleton().findOp(op); in _get_runtime_ops_and_info() local
/external/trusty/arm-trusted-firmware/include/lib/psa/
Drse_crypto_defs.h46 uint32_t op_handle; /* member
/external/pytorch/torch/csrc/jit/mobile/model_tracer/
DTracerRunner.cpp160 const auto op_handle = c10::Dispatcher::singleton().findOp(op); in _get_runtime_ops_and_schema() local
/external/pytorch/torch/csrc/inductor/aoti_torch/
Doss_proxy_executor.cpp341 c10::OperatorHandle op_handle = in OSSProxyExecutor() local
/external/pytorch/aten/src/ATen/cuda/tunable/
DGemmHipblaslt.h486 auto op_handle = at::cuda::getCurrentCUDABlasLtHandle(); in Call() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/
Dqlinear_prepack.cpp342 const std::optional<c10::OperatorHandle> op_handle = in callOpByName() local
/external/pytorch/test/cpp/jit/
Dtest_alias_analysis.cpp1689 const auto& op_handle = c10::Dispatcher::singleton().findOp( in TEST() local
/external/pytorch/aten/src/ATen/test/
Dvulkan_quantized_api_test.cpp139 const std::optional<c10::OperatorHandle> op_handle = in callOpByName() local
Dvulkan_api_test.cpp252 const std::optional<c10::OperatorHandle> op_handle = in callOpByName() local
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc356 void XlaBuilder::IsConstantVisitor(const int64_t op_handle, int depth, in IsConstantVisitor()