Home
last modified time | relevance | path

Searched refs:TFE_ContextSetExecutorForThread (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_experimental_test.cc224 TFE_ContextSetExecutorForThread(ctx, executor); in TEST()
237 TFE_ContextSetExecutorForThread(ctx, executor); in TEST()
279 TFE_ContextSetExecutorForThread(ctx, executor); in TEST()
305 TFE_ContextSetExecutorForThread(ctx, old_executor); in TEST()
331 TFE_ContextSetExecutorForThread(ctx, executor); in Executor_MatMul_CPU()
347 TFE_ContextSetExecutorForThread(ctx, old_executor); in Executor_MatMul_CPU()
H A Dc_api_experimental.h368 TF_CAPI_EXPORT extern void TFE_ContextSetExecutorForThread(TFE_Context*,
H A Dc_api_experimental.cc592 void TFE_ContextSetExecutorForThread(TFE_Context* ctx, TFE_Executor* executor) { in TFE_ContextSetExecutorForThread() function
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/parallel_device/
H A Dparallel_device_lib.cc247 TFE_ContextSetExecutorForThread(context, executor_.get()); in Execute()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1039 pywrap_tfe.TFE_ContextSetExecutorForThread(self._context_handle,
1059 pywrap_tfe.TFE_ContextSetExecutorForThread(self._context_handle, e.handle())
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc960 TFE_ContextSetExecutorForThread(tensorflow::InputTFE_Context(ctx), &exc); in PYBIND11_MODULE()