Home
last modified time | relevance | path

Searched refs:op_or_function_name (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A Deager_operation_builder_jni.cc71 const char* op_or_function_name = env->GetStringUTFChars(name, nullptr); in Java_org_tensorflow_EagerOperationBuilder_allocate() local
73 TFE_Op* op = TFE_NewOp(context, op_or_function_name, status); in Java_org_tensorflow_EagerOperationBuilder_allocate()
74 env->ReleaseStringUTFChars(name, op_or_function_name); in Java_org_tensorflow_EagerOperationBuilder_allocate()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api.h249 const char* op_or_function_name,
300 TFE_Context* ctx, const char* op_or_function_name, const char* attr_name,
H A Dc_api.cc617 TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, in TFE_NewOp() argument
621 status->status = new_op->Reset(op_or_function_name, nullptr); in TFE_NewOp()
690 const char* op_or_function_name, in TFE_OpNameGetAttrType() argument
694 TFE_Op* op = TFE_NewOp(ctx, op_or_function_name, status); in TFE_OpNameGetAttrType()
H A Dc_api_experimental.cc41 void TFE_OpReset(TFE_Op* op_to_reset, const char* op_or_function_name, in TFE_OpReset() argument
47 status->status = op->Reset(op_or_function_name, raw_device_name); in TFE_OpReset()
H A Dc_api_experimental.h33 const char* op_or_function_name,
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc970 [](py::handle& ctx, const char* op_or_function_name, in PYBIND11_MODULE()
977 op_or_function_name, attr_name, in PYBIND11_MODULE()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dpywrap_tfe_src.cc91 TFE_Op* GetOp(TFE_Context* ctx, const char* op_or_function_name, in GetOp() argument
98 tensorflow::unwrap(op.get())->Reset(op_or_function_name, raw_device_name); in GetOp()