Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api.h265 typedef struct TFE_Op TFE_Op; typedef
267 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
271 TF_CAPI_EXPORT extern void TFE_DeleteOp(TFE_Op* op);
273 TF_CAPI_EXPORT extern void TFE_OpSetDevice(TFE_Op* op, const char* device_name,
276 TF_CAPI_EXPORT extern const char* TFE_OpGetDevice(TFE_Op* op,
283 TF_CAPI_EXPORT extern void TFE_OpSetXLACompilation(TFE_Op* op,
286 TF_CAPI_EXPORT extern void TFE_OpAddInput(TFE_Op* op, TFE_TensorHandle* input,
289 TF_CAPI_EXPORT extern void TFE_OpAddInputList(TFE_Op* op,
294 TF_CAPI_EXPORT extern TF_AttrType TFE_OpGetAttrType(TFE_Op* op,
305 TF_CAPI_EXPORT extern void TFE_OpSetAttrString(TFE_Op* op,
[all …]
Dc_api.cc236 tensorflow::Status OpInferSingleInputAttrs(TFE_Op* op, in OpInferSingleInputAttrs()
258 void OpInferSingleTypeInputListAttrs(TFE_Op* op, in OpInferSingleTypeInputListAttrs()
274 void OpInferMixedTypeInputListAttrs(TFE_Op* op, in OpInferMixedTypeInputListAttrs()
292 tensorflow::Status OpInferInputListAttrs(TFE_Op* op, TFE_TensorHandle** inputs, in OpInferInputListAttrs()
562 TFE_Op* TFE_NewOp(TFE_Context* ctx, const char* op_or_function_name, in TFE_NewOp()
577 return new TFE_Op(ctx, name, false, types, in TFE_NewOp()
590 return new TFE_Op(ctx, name, true, types, nullptr); in TFE_NewOp()
593 void TFE_DeleteOp(TFE_Op* op) { delete op; } in TFE_DeleteOp()
595 void TFE_OpSetDevice(TFE_Op* op, const char* device_name, TF_Status* status) { in TFE_OpSetDevice()
599 const char* TFE_OpGetDevice(TFE_Op* op, TF_Status* status) { in TFE_OpGetDevice()
[all …]
Dc_api_test_util.h44 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b);
47 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a);
53 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input,
Dc_api_test_util.cc116 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp()
119 TFE_Op* op = TFE_NewOp(ctx, "MatMul", status); in MatMulOp()
131 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp()
134 TFE_Op* op = TFE_NewOp(ctx, "Shape", status); in ShapeOp()
158 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, in MinOp()
162 TFE_Op* op = TFE_NewOp(ctx, "Min", status); in MinOp()
Dc_api_internal.h111 struct TFE_Op { struct
112 TFE_Op(TFE_Context* ctx, const char* op, bool is_function, in TFE_Op() argument
136 void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op, argument
Dc_api_test.cc49 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_InitOp()
71 TFE_Op* matmul = MatMulOp(ctx, m, m); in BM_Execute()
170 TFE_Op* matmul = MatMulOp(ctx, h0_task1, h1_task1); in TestRemoteExecute()
252 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h1_task2); in TestRemoteExecuteSilentCopies()
323 TFE_Op* matmul = MatMulOp(ctx, h0_task0, h0_task0); in CheckRemoteMatMulExecutesOK()
673 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopy()
719 TFE_Op* matmul = MatMulOp(ctx, hcpu, hgpu); in TensorHandleSilentCopyLocal()
750 TFE_Op* matmul = MatMulOp(ctx, m, m); in SetAndGetOpDevices()
842 TFE_Op* shape_op = ShapeOp(ctx, hgpu); in TEST()
882 TFE_Op* matmul = MatMulOp(ctx, m, m); in Execute_MatMul_CPU()
[all …]
Dc_api_experimental.h25 TF_CAPI_EXPORT extern void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h,
Dc_api_experimental.cc25 void TFE_OpConsumeInput(TFE_Op* op, TFE_TensorHandle* h, TF_Status* status) { in TFE_OpConsumeInput()
Dc_api_experimental_test.cc52 TFE_Op* matmul = MatMulOp(ctx, m, m); in ExecuteWithProfiling()
/external/tensorflow/tensorflow/c/
Dc_api_experimental_test.cc188 TFE_Op* matmul_op = MatMulOp(ctx, m, m); in TEST()
232 TFE_Op* send_op = TFE_NewOp(ctx, "_Send", status); in TEST()
251 TFE_Op* recv_op = TFE_NewOp(ctx, "_Recv", status); in TEST()
344 void AddEagerOpToGraphAndCheck(TFE_Op* op, Callable checker) { in AddEagerOpToGraphAndCheck()
367 TFE_Op* op = MatMulOp(eager_ctx_, m, m); in TEST_F()
400 TFE_Op* op = MinOp(eager_ctx_, axis, axis); in TEST_F()
424 TFE_Op* squeeze = TFE_NewOp(eager_ctx_, "Squeeze", status_); in TEST_F()
451 TFE_Op* identityn = TFE_NewOp(eager_ctx_, "IdentityN", status_); in TEST_F()
Dc_api_experimental.h193 TF_CAPI_EXPORT extern void TFE_OpPrintDebugString(TFE_Op* op);
206 TFE_Op* op, TFE_TensorHandle** retvals, int* num_retvals,
300 TFE_Op* op, TFE_TraceContext* trace_ctx, TFE_TensorHandle** retvals,
Dc_api_experimental.cc8572 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> queueOp( in createTFEQueue()
8603 TFE_Op* op = TFE_NewOp(ctx, "QueueEnqueueV2", status); in createTFEEnqueue()
8605 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op_deleter(op, TFE_DeleteOp); in createTFEEnqueue()
8625 TFE_Op* op = TFE_NewOp(ctx, "QueueDequeueV2", status); in createTFEDequeue()
8627 std::unique_ptr<TFE_Op, decltype(&TFE_DeleteOp)> op_deleter(op, TFE_DeleteOp); in createTFEDequeue()
8766 void TFE_OpPrintDebugString(TFE_Op* op) { in TFE_OpPrintDebugString()
8778 TFE_ExecuteOpNotification* TFE_ExecuteOpInNewThread(TFE_Op* op, in TFE_ExecuteOpInNewThread()
9056 TF_Operation* TFE_AddEagerOpToGraph(TFE_Op* op, TFE_TraceContext* trace_ctx, in TFE_AddEagerOpToGraph()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe_src.cc267 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_list, in SetOpAttrList()
374 std::unique_ptr<const TFE_Op*[]> funcs(new const TFE_Op*[num_values]); in SetOpAttrList()
418 TFE_Op* GetFunc(TFE_Context* ctx, const tensorflow::NameAttrList& func, in GetFunc()
420 TFE_Op* func_op = TFE_NewOp(ctx, func.name().data(), status); in GetFunc()
430 TFE_Context* ctx, TFE_Op* op, const tensorflow::OpDef::AttrDef& attr, in SetOpAttrListDefault()
512 std::unique_ptr<const TFE_Op*[]> funcs(new const TFE_Op*[num_values]); in SetOpAttrListDefault()
525 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_value, in SetOpAttrScalar()
622 TFE_Context* ctx, TFE_Op* op, const tensorflow::AttrValue& default_value, in SetOpAttrScalarDefault()
634 void SetOpAttrs(TFE_Context* ctx, TFE_Op* op, PyObject* attrs, int start_index, in SetOpAttrs()
671 TFE_Context* ctx, TFE_Op* op, const tensorflow::OpDef::AttrDef& attr, in SetOpAttrWithDefaults()
[all …]
Dpywrap_tensor.cc182 TFE_Op* op = TFE_NewOp(ctx, op_name, out_status); in EagerCast()