Home
last modified time | relevance | path

Searched refs:TFE_Op (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api.h246 typedef struct TFE_Op TFE_Op; typedef
248 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
252 TF_CAPI_EXPORT extern void TFE_DeleteOp(TFE_Op* op);
254 TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
257 TF_CAPI_EXPORT extern const char* TFE_OpGetDevice(TFE_Op* op,
264 TF_CAPI_EXPORT extern void TFE_OpSetXLACompilation(TFE_Op* op,
267 TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input,
270 TF_CAPI_EXPORT extern void TFE_OpAddInputList(TFE_Op* op,
275 TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op,
286 TF_CAPI_EXPORT extern void TFE_OpSetAttrString(TFE_Op* op,
[all …]
Dc_api_test_util.h46 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
49 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a);
52 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
58 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input,
Dc_api_test_util.cc134 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp()
137 TFE_Op* op = TFE_NewOp(ctx, "MatMul", status); in MatMulOp()
149 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { in IdentityOp()
152 TFE_Op* op = TFE_NewOp(ctx, "Identity", status); in IdentityOp()
162 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp()
165 TFE_Op* op = TFE_NewOp(ctx, "Shape", status); in ShapeOp()
189 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, in MinOp()
193 TFE_Op* op = TFE_NewOp(ctx, "Min", status); in MinOp()
Dc_api.cc98 const tensorflow::OpDef* GetOpDef(TFE_Op* op, TF_Status* status) { in GetOpDef()
1206 TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, in TFE_NewOp()
1208 std::unique_ptr<TFE_Op> new_op( in TFE_NewOp()
1209 new TFE_Op{tensorflow::EagerOperation(ctx->context)}); in TFE_NewOp()
1218 void TFE_DeleteOp(TFE_Op* op) { delete op; } in TFE_DeleteOp()
1220 void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { in TFE_OpSetDevice()
1224 const char* TFE_OpGetDevice(TFE_Op* op, TF_Status* status) { in TFE_OpGetDevice()
1231 void TFE_OpSetXLACompilation(TFE_Op* op, unsigned char enable) { in TFE_OpSetXLACompilation()
1239 void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input, TF_Status* status) { in TFE_OpAddInput()
1248 void TFE_OpAddInputList(TFE_Op* op, TFE_TensorHandle** inputs, int num_inputs, in TFE_OpAddInputList()
[all …]
Dc_api_test.cc50 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp()
72 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute()
107 TFE_Op* identity = IdentityOp(ctx, m); in BM_Execute_Identity()
387 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy()
435 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopyLocal()
468 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices()
560 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST()
602 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU()
639 TFE_Op* matmul = MatMulOp(ctx, m1, m2); in Execute_MatMul_CPU_Runtime_Error()
643 TFE_Op* matmul2 = MatMulOp(ctx, m1, m1); in Execute_MatMul_CPU_Runtime_Error()
[all …]
Dc_api_experimental.h32 TF_CAPI_EXPORT extern void TFE_OpReset(TFE_Op* op_to_reset,
37 TF_CAPI_EXPORT extern void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h,
368 TFE_Op* op, TFE_CancellationManager* cancellation_manager,
Dc_api_remote_test.cc87 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute()
171 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h1_task2); in TestRemoteExecuteSilentCopies()
260 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
315 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); in CheckRemoteMatMulExecutesOK()
Dc_api_internal.h92 struct TFE_Op { struct
222 void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
Dc_api_experimental_test.cc52 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithProfiling()
341 TFE_Op* op = TFE_NewOp(ctx, "ident", status); in TEST()
417 TFE_Op* op = TFE_NewOp(ctx, "ident", status); in TEST()
465 TFE_Op* matmul = MatMulOp(ctx, m, m); in Executor_MatMul_CPU()
Dc_api_experimental.cc33 void TFE_OpReset(TFE_Op* op_to_reset, const char* op_or_function_name, in TFE_OpReset()
44 void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h, TF_Status* status) { in TFE_OpConsumeInput()
589 void TFE_OpSetCancellationManager(TFE_Op* op, in TFE_OpSetCancellationManager()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_operation_builder_jni.cc30 TFE_Op* requireOp(JNIEnv* env, jlong handle) { in requireOp()
36 return reinterpret_cast<TFE_Op*>(handle); in requireOp()
73 TFE_Op* op = TFE_NewOp(context, op_or_function_name, status); in Java_org_tensorflow_EagerOperationBuilder_allocate()
80 static_assert(sizeof(jlong) >= sizeof(TFE_Op*), in Java_org_tensorflow_EagerOperationBuilder_allocate()
88 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(op_handle)); in Java_org_tensorflow_EagerOperationBuilder_delete()
93 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_execute()
118 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_setDevice()
130 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_addInput()
142 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_addInputList()
167 TFE_Op* op = requireOp(env, op_handle); in Java_org_tensorflow_EagerOperationBuilder_setAttrString()
[all …]
Deager_operation_jni.cc30 TFE_Op* requireOp(JNIEnv* env, jlong handle) { in requireOp()
36 return reinterpret_cast<TFE_Op*>(handle); in requireOp()
53 TFE_DeleteOp(reinterpret_cast<TFE_Op*>(handle)); in Java_org_tensorflow_EagerOperation_delete()
80 TFE_Op* op = requireOp(env, handle); in Java_org_tensorflow_EagerOperation_outputListLength()
96 TFE_Op* op = requireOp(env, handle); in Java_org_tensorflow_EagerOperation_inputListLength()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc96 TFE_Op* matmul_op = MatMulOp(ctx, m, m); in TEST()
140 TFE_Op* send_op = TFE_NewOp(ctx, "_Send", status); in TEST()
159 TFE_Op* recv_op = TFE_NewOp(ctx, "_Recv", status); in TEST()
224 TFE_Op* op, in CheckOutputShapes()
285 TFE_Op* matmul_op; in TEST_F()
325 TFE_Op* reshape_op = TFE_NewOp(tfe_context_, "Reshape", status_); in TEST_F()
336 TFE_Op* fill_op = TFE_NewOp(tfe_context_, "Fill", status_); in TEST_F()
Dc_api_experimental.h195 TF_CAPI_EXPORT extern void TFE_OpPrintDebugString(TFE_Op* op);
208 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
349 TFE_Op* op, TF_ShapeAndTypeList* input_shapes, TF_Tensor** input_tensors,
Dc_api_experimental.cc351 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> queueOp( in createTFEQueue()
382 TFE_Op* op = TFE_NewOp(ctx, "QueueEnqueueV2", status); in createTFEEnqueue()
384 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op_deleter(op, TFE_DeleteOp); in createTFEEnqueue()
404 TFE_Op* op = TFE_NewOp(ctx, "QueueDequeueV2", status); in createTFEDequeue()
406 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op_deleter(op, TFE_DeleteOp); in createTFEDequeue()
535 void TFE_OpPrintDebugString(TFE_Op* op) { in TFE_OpPrintDebugString()
547 TFE_ExecuteOpNotification* TFE_ExecuteOpInNewThread(TFE_Op* op, in TFE_ExecuteOpInNewThread()
868 void TFE_InferShapes(TFE_Op* tfe_op, TF_ShapeAndTypeList* input_shapes, in TFE_InferShapes()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc59 thread_local std::map<TFE_Context*, std::unique_ptr<TFE_Op>>
64 std::unique_ptr<TFE_Op> ReleaseThreadLocalOp(TFE_Context* ctx) { in ReleaseThreadLocalOp()
72 TFE_Op* GetOp(TFE_Context* ctx, const char* op_or_function_name, in GetOp()
74 std::unique_ptr<TFE_Op> op = ReleaseThreadLocalOp(ctx); in GetOp()
76 op.reset(new TFE_Op{tensorflow::EagerOperation(ctx->context)}); in GetOp()
86 void ReturnOp(TFE_Context* ctx, TFE_Op* op) { in ReturnOp()
357 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_list, in SetOpAttrList()
464 std::unique_ptr<const TFE_Op*[]> funcs(new const TFE_Op*[num_values]); in SetOpAttrList()
508 TFE_Op* GetFunc(TFE_Context* ctx, const tensorflow::NameAttrList& func, in GetFunc()
510 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
[all …]
Dpywrap_tensor.cc205 TFE_Op* op = TFE_NewOp(ctx, op_name, out_status); in EagerCast()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.cc109 TFE_Op* op = TFE_NewOp(context, node_def->op().c_str(), status); in EvaluateOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dconstant-fold.mlir190 // Ops with unimplemented attributes which couldn't be added to the TFE_Op.