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 D | eager_operation_builder_jni.cc | 71 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 D | c_api.h | 249 const char* op_or_function_name, 300 TFE_Context* ctx, const char* op_or_function_name, const char* attr_name,
|
H A D | c_api.cc | 617 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 D | c_api_experimental.cc | 41 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 D | c_api_experimental.h | 33 const char* op_or_function_name,
|
/aosp_15_r20/external/tensorflow/tensorflow/python/ |
H A D | tfe_wrapper.cc | 970 [](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 D | pywrap_tfe_src.cc | 91 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()
|