Home
last modified time | relevance | path

Searched refs:TFE_ContextGetExecutorForThread (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_experimental_test.cc276 TFE_Executor* old_executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
328 TFE_Executor* old_executor = TFE_ContextGetExecutorForThread(ctx); in Executor_MatMul_CPU()
480 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
514 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
H A Dc_api_cluster_test.cc84 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in CheckRemoteMatMulExecutesOK()
138 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestRemoteExecuteChangeServerDef()
H A Dc_api_remote_test.cc105 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestRemoteExecute()
H A Dc_api_test.cc102 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in BM_Execute()
138 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in BM_Execute_Identity()
297 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TensorHandleCopyBetweenDevicesError()
453 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TensorHandleSilentCopy()
597 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
818 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in Execute_MatMul_CPU_Runtime_Error()
1301 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in BM_ExecuteFunction()
H A Dc_api_remote_test_util.cc222 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestRemoteExecuteSilentCopies()
H A Dc_api_experimental.h372 TF_CAPI_EXPORT extern TFE_Executor* TFE_ContextGetExecutorForThread(
H A Dc_api_distributed_test.cc225 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestFunctionWithPackedInput()
H A Dc_api_experimental.cc596 TFE_Executor* TFE_ContextGetExecutorForThread(TFE_Context* ctx) { in TFE_ContextGetExecutorForThread() function
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/integration_test/
H A Dc_api_coordination_test.cc142 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
252 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
481 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST_P()
H A Dc_api_multi_client_test.cc77 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in StartWorkers()
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1054 pywrap_tfe.TFE_ContextGetExecutorForThread(self._context_handle))
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc965 return TFE_ContextGetExecutorForThread(tensorflow::InputTFE_Context(o)); in PYBIND11_MODULE()