Home
last modified time | relevance | path

Searched refs:SetExecutorForThread (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/
H A Dpy_func.cc206 tensorflow::unwrap(ctx)->SetExecutorForThread(new_executor.get()); in DoCallPyFunc()
243 tensorflow::unwrap(ctx)->SetExecutorForThread(old_executor); in DoCallPyFunc()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dimmediate_execution_context.h209 virtual void SetExecutorForThread(EagerExecutor* executor) = 0;
H A Dc_api_experimental.cc593 tensorflow::unwrap(ctx)->SetExecutorForThread(executor->executor()); in TFE_ContextSetExecutorForThread()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.h208 void SetExecutorForThread(tensorflow::EagerExecutor* executor) override;
H A Dc_api_tfrt.cc1235 void ContextInterface::SetExecutorForThread( in SetExecutorForThread() function in tfrt::tf::ContextInterface
1237 GetEagerContext()->SetExecutorForThread(executor); in SetExecutorForThread()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext.h174 void SetExecutorForThread(EagerExecutor* executor) override;
H A Dcontext.cc430 void EagerContext::SetExecutorForThread(EagerExecutor* executor) { in SetExecutorForThread() function in tensorflow::EagerContext