Home
last modified time | relevance | path

Searched refs:TFE_Context (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api.h91 typedef struct TFE_Context TFE_Context; typedef
93 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContext(
95 TF_CAPI_EXPORT extern void TFE_DeleteContext(TFE_Context* ctx);
96 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
101 TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx,
109 TFE_Context*, TFE_ContextDevicePlacementPolicy);
114 TFE_ContextGetDevicePlacementPolicy(TFE_Context*);
117 TF_CAPI_EXPORT extern void TFE_ContextSetAsyncForThread(TFE_Context*,
127 TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx,
140 TF_CAPI_EXPORT extern void TFE_ContextAsyncWait(TFE_Context*,
[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,
59 bool GetDeviceName(TFE_Context* ctx, tensorflow::string* device_name,
Dc_api.cc158 TFE_Context* ctx) { in UpdateTFE_ContextWithServerDef()
328 TF_CAPI_EXPORT extern void TFE_ContextSetAsyncForThread(TFE_Context* ctx, in TFE_ContextSetAsyncForThread()
336 TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) { in TFE_NewContext()
348 return new TFE_Context(opts->session_options.options, opts->policy, in TFE_NewContext()
353 TFE_Context* TFE_NewContextFromSession(const TFE_ContextOptions* opts, in TFE_NewContextFromSession()
360 return new TFE_Context(opts->session_options.options, opts->policy, in TFE_NewContextFromSession()
365 void TFE_DeleteContext(TFE_Context* ctx) { delete ctx; } in TFE_DeleteContext()
367 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
376 void TFE_ContextClearCaches(TFE_Context* ctx, TF_Status* status) { in TFE_ContextClearCaches()
381 TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx, in TFE_ContextSetServerDef()
[all …]
Dc_api_experimental.h43 TF_CAPI_EXPORT extern void TFE_ProfilerSerializeToString(TFE_Context* ctx,
53 TFE_ProfilerContext* profiler_context, TFE_Context* eager_context);
72 TF_CAPI_EXPORT extern void TFE_ContextEnableGraphCollection(TFE_Context* ctx);
76 TF_CAPI_EXPORT extern void TFE_ContextDisableGraphCollection(TFE_Context* ctx);
Dc_api_experimental.cc39 void TFE_ProfilerSerializeToString(TFE_Context* ctx, TFE_Profiler* profiler, in TFE_ProfilerSerializeToString()
59 TFE_Context* eager_context) { in TFE_ProfilerContextSetEagerContext()
73 void TFE_ContextEnableGraphCollection(TFE_Context* ctx) { in TFE_ContextEnableGraphCollection()
77 void TFE_ContextDisableGraphCollection(TFE_Context* ctx) { in TFE_ContextDisableGraphCollection()
Dc_api_internal.h65 struct TFE_Context { struct
66 TFE_Context(const tensorflow::SessionOptions& opts, in TFE_Context() argument
112 TFE_Op(TFE_Context* ctx, const char* op, bool is_function, in TFE_Op()
136 void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
Dc_api_test.cc42 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_InitOp()
66 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute()
94 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
152 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecute()
235 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteSilentCopies()
317 void CheckRemoteMatMulExecutesOK(TFE_Context* ctx, in CheckRemoteMatMulExecutesOK()
367 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteChangeServerDef()
464 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevices()
541 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevicesError()
578 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
[all …]
Dc_api_test_util.cc116 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp()
131 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp()
158 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, in MinOp()
176 bool GetDeviceName(TFE_Context* ctx, string* device_name, in GetDeviceName()
Dc_api_experimental_test.cc43 TFE_Context* ctx = TFE_NewContext(opts, status); in ExecuteWithProfiling()
110 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.h148 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContextFromSession(
160 TF_CAPI_EXPORT extern TFE_Context* TFE_CreateContextFromSession(
169 TFE_Context* ctx, int tensor_id, TF_DataType inputType, TF_Status* status);
177 TFE_Context* ctx, int tensor_id, TFE_TensorHandle* tensor,
265 TF_CAPI_EXPORT extern void TFE_EnableCollectiveOps(TFE_Context* ctx,
Dc_api_experimental_test.cc182 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
224 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
360 TFE_Context* eager_ctx_;
Dc_api_experimental.cc8544 TFE_Context* TFE_CreateContextFromSession(TF_Session* session, in TFE_CreateContextFromSession()
8570 static TFE_TensorHandle* createTFEQueue(TFE_Context* ctx, TF_DataType inputType, in createTFEQueue()
8600 static void createTFEEnqueue(TFE_Context* ctx, TF_DataType inputType, in createTFEEnqueue()
8621 static TFE_TensorHandle* createTFEDequeue(TFE_Context* ctx, in createTFEDequeue()
8651 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_DequeueNamedTensor()
8663 TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_DequeueNamedTensorFromCtx()
8683 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_EnqueueNamedTensor()
8695 void TFE_EnqueueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_EnqueueNamedTensorFromCtx()
8715 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_EnqueueVariantTensor()
8732 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_DequeueVariantTensor()
[all …]
/external/tensorflow/tensorflow/python/
Dpywrap_tfe.i181 %typemap(in) (TFE_Context*) {
182 $1 = (TFE_Context*)PyCapsule_GetPointer($input, nullptr);
185 %typemap(out) (TFE_Context*) {
302 %typemap(in) (TFE_Context*);
303 %typemap(out) (TFE_Context*);
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tensor.h34 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle,
Dpywrap_tensor.cc43 TFE_Context* GetContext(PyObject* ctx) { in GetContext()
44 TFE_Context* context = in GetContext()
45 reinterpret_cast<TFE_Context*>(PyCapsule_GetPointer(ctx, nullptr)); in GetContext()
91 TFE_Context* context = GetContext(ctx); in CopyToDevice()
176 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle, in EagerCast()
Dpywrap_tfe.h49 void TFE_Py_Execute(TFE_Context* ctx, const char* device_name,
Dpywrap_tfe_src.cc97 TFE_Context* ctx;
267 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_list, in SetOpAttrList()
418 TFE_Op* GetFunc(TFE_Context* ctx, const tensorflow::NameAttrList& func, in GetFunc()
430 TFE_Context* ctx, TFE_Op* op, const tensorflow::OpDef::AttrDef& attr, 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()
714 void TFE_Py_Execute(TFE_Context* ctx, const char* device_name, in TFE_Py_Execute()
894 TFE_Context* ctx = in TFE_DeleteContextCapsule()
[all …]
/external/tensorflow/tensorflow/go/
Dcontext.go74 c *C.TFE_Context