Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/c/eager/
Dc_api.h97 typedef struct TFE_Context TFE_Context; typedef
99 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContext(
101 TF_CAPI_EXPORT extern void TFE_DeleteContext(TFE_Context* ctx);
102 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
107 TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx);
114 TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy);
119 TFE_ContextGetDevicePlacementPolicy(TFE_Context* ctx);
127 TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx,
193 TFE_TensorHandle* h, TFE_Context* ctx, const char* device_name,
248 TF_CAPI_EXPORT extern TFE_Op* TFE_NewOp(TFE_Context* ctx,
[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,
64 bool GetDeviceName(TFE_Context* ctx, tensorflow::string* device_name,
Dc_api_experimental.h68 TF_CAPI_EXPORT extern void TFE_ContextEnableGraphCollection(TFE_Context* ctx);
72 TF_CAPI_EXPORT extern void TFE_ContextDisableGraphCollection(TFE_Context* ctx);
340 TFE_Context*, TFE_ContextMirroringPolicy);
345 TFE_Context*);
409 TF_CAPI_EXPORT extern void TFE_ContextSetExecutorForThread(TFE_Context*,
414 TFE_Context*);
425 TF_CAPI_EXPORT extern void TFE_ContextUpdateServerDef(TFE_Context* ctx,
433 TF_CAPI_EXPORT extern bool TFE_ContextCheckAlive(TFE_Context* ctx,
456 TFE_Context* ctx, const char* device_name, TF_DataType, const int64_t* dims,
463 TF_CAPI_EXPORT extern void TFE_HostAddressSpace(TFE_Context* ctx,
Dc_api.cc269 TFE_Context* ctx, const std::vector<string>& remote_workers, in CreateRemoteContexts()
344 TFE_Context* ctx, const std::vector<string>& remote_workers, in UpdateRemoteContexts()
448 TFE_Context* ctx, bool reset_context) { in UpdateTFE_ContextWithServerDef()
698 TFE_Context* TFE_NewContext(const TFE_ContextOptions* opts, TF_Status* status) { in TFE_NewContext()
710 return new TFE_Context{new tensorflow::EagerContext( in TFE_NewContext()
720 TFE_Context* TFE_NewContextFromSession(const TFE_ContextOptions* opts, in TFE_NewContextFromSession()
728 return new TFE_Context{new tensorflow::EagerContext( in TFE_NewContextFromSession()
738 void TFE_DeleteContext(TFE_Context* ctx) { in TFE_DeleteContext()
746 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
752 void TFE_ContextClearCaches(TFE_Context* ctx) { in TFE_ContextClearCaches()
[all …]
Dc_api_test.cc43 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_InitOp()
67 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute()
102 TFE_Context* ctx = TFE_NewContext(opts, status); in BM_Execute_Identity()
133 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
175 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevices()
252 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenDevicesError()
292 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
372 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleSilentCopy()
418 TFE_Context* ctx = TFE_NewContext(opts, status.get()); in TensorHandleSilentCopyLocal()
463 TFE_Context* ctx = TFE_NewContext(opts, status); in SetAndGetOpDevices()
[all …]
Dc_api_test_util.cc134 TFE_Op* MatMulOp(TFE_Context* ctx, TFE_TensorHandle* a, TFE_TensorHandle* b) { in MatMulOp()
149 TFE_Op* IdentityOp(TFE_Context* ctx, TFE_TensorHandle* a) { in IdentityOp()
162 TFE_Op* ShapeOp(TFE_Context* ctx, TFE_TensorHandle* a) { in ShapeOp()
189 TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, in MinOp()
207 bool GetDeviceName(TFE_Context* ctx, string* device_name, in GetDeviceName()
Dc_api_experimental.cc77 void TFE_ContextEnableGraphCollection(TFE_Context* ctx) { in TFE_ContextEnableGraphCollection()
81 void TFE_ContextDisableGraphCollection(TFE_Context* ctx) { in TFE_ContextDisableGraphCollection()
551 TFE_Context* ctx, TFE_ContextMirroringPolicy policy) { in TFE_ContextSetThreadLocalMirroringPolicy()
560 TFE_Context* ctx) { in TFE_ContextGetMirroringPolicy()
615 void TFE_ContextSetExecutorForThread(TFE_Context* ctx, TFE_Executor* executor) { in TFE_ContextSetExecutorForThread()
619 TFE_Executor* TFE_ContextGetExecutorForThread(TFE_Context* ctx) { in TFE_ContextGetExecutorForThread()
623 void TFE_HostAddressSpace(TFE_Context* ctx, TF_Buffer* buf) { in TFE_HostAddressSpace()
Dc_api_remote_test.cc69 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecute()
154 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteSilentCopies()
240 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteDeleteContextWithOutstandingRPC()
309 void CheckRemoteMatMulExecutesOK(TFE_Context* ctx, in CheckRemoteMatMulExecutesOK()
361 TFE_Context* ctx = TFE_NewContext(opts, status); in TestRemoteExecuteChangeServerDef()
Dc_api_internal.h65 struct TFE_Context { struct
222 void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op,
Dc_api_experimental_test.cc46 TFE_Context* ctx = TFE_NewContext(opts, status); in ExecuteWithProfiling()
321 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
397 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
456 TFE_Context* ctx = TFE_NewContext(opts, status); in Executor_MatMul_CPU()
505 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
/external/tensorflow/tensorflow/java/src/main/native/
Deager_session_jni.cc44 TFE_Context* context = TFE_NewContext(opts, status); in Java_org_tensorflow_EagerSession_allocate()
54 static_assert(sizeof(jlong) >= sizeof(TFE_Context*), in Java_org_tensorflow_EagerSession_allocate()
63 TFE_DeleteContext(reinterpret_cast<TFE_Context*>(handle)); in Java_org_tensorflow_EagerSession_delete()
Deager_operation_builder_jni.cc39 TFE_Context* requireContext(JNIEnv* env, jlong handle) { in requireContext()
44 return reinterpret_cast<TFE_Context*>(handle); in requireContext()
69 TFE_Context* context = requireContext(env, context_handle); in Java_org_tensorflow_EagerOperationBuilder_allocate()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.h150 TF_CAPI_EXPORT extern TFE_Context* TFE_NewContextFromSession(
162 TF_CAPI_EXPORT extern TFE_Context* TFE_CreateContextFromSession(
171 TFE_Context* ctx, int tensor_id, TF_DataType inputType, TF_Status* status);
179 TFE_Context* ctx, int tensor_id, TFE_TensorHandle* tensor,
295 TF_CAPI_EXPORT extern void TFE_EnableCollectiveOps(TFE_Context* ctx,
Dc_api_experimental.cc323 TFE_Context* TFE_CreateContextFromSession(TF_Session* session, in TFE_CreateContextFromSession()
349 static TFE_TensorHandle* createTFEQueue(TFE_Context* ctx, TF_DataType inputType, in createTFEQueue()
379 static void createTFEEnqueue(TFE_Context* ctx, TF_DataType inputType, in createTFEEnqueue()
400 static TFE_TensorHandle* createTFEDequeue(TFE_Context* ctx, in createTFEDequeue()
430 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_DequeueNamedTensor()
442 TFE_TensorHandle* TFE_DequeueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_DequeueNamedTensorFromCtx()
462 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_EnqueueNamedTensor()
474 void TFE_EnqueueNamedTensorFromCtx(TFE_Context* ctx, int tensor_id, in TFE_EnqueueNamedTensorFromCtx()
494 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_EnqueueVariantTensor()
511 std::unique_ptr<TFE_Context, decltype(&TFE_DeleteContext)> ctx_deleter( in TFE_DequeueVariantTensor()
[all …]
Dc_api_experimental_test.cc90 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
132 TFE_Context* ctx = TFE_NewContext(opts, status); in TEST()
281 TFE_Context* tfe_context_;
/external/tensorflow/tensorflow/python/lib/core/
Dpy_seq_tensor.h37 TFE_TensorHandle* PySeqToTFE_TensorHandle(TFE_Context* ctx, PyObject* obj,
Dpy_func.cc190 TFE_Context* ctx = nullptr; in DoCallPyFunc()
195 ctx = reinterpret_cast<TFE_Context*>(PyCapsule_GetPointer( in DoCallPyFunc()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe.h49 void TFE_Py_Execute(TFE_Context* ctx, const char* device_name,
60 void TFE_Py_ExecuteCancelable(TFE_Context* ctx, const char* device_name,
Dpywrap_tensor.h36 TFE_TensorHandle* ConvertToEagerTensor(TFE_Context* ctx, PyObject* value,
Dpywrap_tensor.cc47 TFE_Context* GetContextHandle(PyObject* py_context) { in GetContextHandle()
61 auto* ctx = reinterpret_cast<TFE_Context*>( in GetContextHandle()
199 TFE_TensorHandle* EagerCast(TFE_Context* ctx, TFE_TensorHandle* handle, in EagerCast()
233 TFE_TensorHandle* ConvertToEagerTensorUncached(TFE_Context* ctx, in ConvertToEagerTensorUncached()
324 TFE_TensorHandle* ConvertToEagerTensor(TFE_Context* ctx, PyObject* value, in ConvertToEagerTensor()
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()
86 void ReturnOp(TFE_Context* ctx, TFE_Op* op) { in ReturnOp()
187 TFE_Context* ctx;
357 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_list, in SetOpAttrList()
508 TFE_Op* GetFunc(TFE_Context* ctx, const tensorflow::NameAttrList& func, in GetFunc()
520 TFE_Context* ctx, TFE_Op* op, const tensorflow::OpDef::AttrDef& attr, in SetOpAttrListDefault()
615 TFE_Context* ctx, TFE_Op* op, const char* key, PyObject* py_value, in SetOpAttrScalar()
712 TFE_Context* ctx, TFE_Op* op, const tensorflow::AttrValue& default_value, in SetOpAttrScalarDefault()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Deval_util.h33 TFE_Context* context, llvm::SmallVectorImpl<mlir::Attribute>* results);
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_fold.cc49 static TFE_Context* ctx = initialize(); in ConstantFoldFallbackHook()
/external/tensorflow/tensorflow/go/
Dcontext.go74 c *C.TFE_Context
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc84 TFE_Context* InputTFE_Context(const py::handle& ctx) { in InputTFE_Context()
85 return static_cast<TFE_Context*>(PyCapsule_GetPointer(ctx.ptr(), nullptr)); in InputTFE_Context()
88 PyObject* OutputTFE_Context(TFE_Context* context) { in OutputTFE_Context()
219 TFE_Context* ctx = tensorflow::InputTFE_Context(context); in InputTFE_OutputTensorHandles()
273 py::class_<TFE_Context> TFE_Context_class(m, "TFE_Context"); in PYBIND11_MODULE()
349 TFE_Context* context = TFE_NewContext(opts, status.get()); in PYBIND11_MODULE()

12