Searched refs:TFE_Executor (Results 1 – 11 of 11) sorted by relevance
22 struct TFE_Executor { struct23 explicit TFE_Executor(bool async) in TFE_Executor() function26 explicit TFE_Executor(tensorflow::EagerExecutor* executor) in TFE_Executor() argument
302 typedef struct TFE_Executor TFE_Executor; typedef307 TF_CAPI_EXPORT extern TFE_Executor* TFE_NewExecutor(bool is_async);312 TF_CAPI_EXPORT extern void TFE_DeleteExecutor(TFE_Executor*);315 TF_CAPI_EXPORT extern bool TFE_ExecutorIsAsync(TFE_Executor*);325 TFE_Executor*, TF_Status* status);334 TF_CAPI_EXPORT extern void TFE_ExecutorClearError(TFE_Executor*);339 TFE_Executor*);342 TF_CAPI_EXPORT extern TFE_Executor* TFE_ContextGetExecutorForThread(
517 TFE_Executor* TFE_NewExecutor(bool is_async) { in TFE_NewExecutor()518 return new TFE_Executor(is_async); in TFE_NewExecutor()521 void TFE_DeleteExecutor(TFE_Executor* executor) { delete executor; } in TFE_DeleteExecutor()523 bool TFE_ExecutorIsAsync(TFE_Executor* executor) { in TFE_ExecutorIsAsync()527 void TFE_ExecutorWaitForAllPendingNodes(TFE_Executor* executor, in TFE_ExecutorWaitForAllPendingNodes()532 void TFE_ExecutorClearError(TFE_Executor* executor) { in TFE_ExecutorClearError()536 void TFE_ContextSetExecutorForThread(TFE_Context* ctx, TFE_Executor* executor) { in TFE_ContextSetExecutorForThread()540 TFE_Executor* TFE_ContextGetExecutorForThread(TFE_Context* ctx) { in TFE_ContextGetExecutorForThread()541 return new TFE_Executor(&tensorflow::unwrap(ctx)->Executor()); in TFE_ContextGetExecutorForThread()
223 TFE_Executor* executor = TFE_NewExecutor(/*is_async=*/false); in TEST()235 TFE_Executor* executor = TFE_NewExecutor(/*is_async=*/false); in TEST()275 TFE_Executor* old_executor = TFE_ContextGetExecutorForThread(ctx); in TEST()276 TFE_Executor* executor = TFE_NewExecutor(async); in TEST()326 TFE_Executor* old_executor = TFE_ContextGetExecutorForThread(ctx); in Executor_MatMul_CPU()327 TFE_Executor* executor = TFE_NewExecutor(async); in Executor_MatMul_CPU()477 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()511 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()
105 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestRemoteExecute()
222 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestRemoteExecuteSilentCopies()
84 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in CheckRemoteMatMulExecutesOK()138 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestRemoteExecuteChangeServerDef()
101 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in BM_Execute()140 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in BM_Execute_Identity()299 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TensorHandleCopyBetweenDevicesError()455 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TensorHandleSilentCopy()599 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TEST()790 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in Execute_MatMul_CPU_Runtime_Error()1184 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in BM_ExecuteFunction()
225 TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx); in TestFunctionWithPackedInput()
46 void operator()(TFE_Executor* to_delete) const { in operator ()()51 using ExecutorPtr = std::unique_ptr<TFE_Executor, ExecutorDeleter>;
50 PYBIND11_MAKE_OPAQUE(TFE_Executor);468 py::class_<TFE_Executor> TFE_Executor_class(m, "TFE_Executor"); in PYBIND11_MODULE()783 TFE_Executor* exc = TFE_NewExecutor(is_async); in PYBIND11_MODULE()789 m.def("TFE_ExecutorWaitForAllPendingNodes", [](TFE_Executor& exc) { in PYBIND11_MODULE()800 TFE_Executor& exc) { in PYBIND11_MODULE()