Home
last modified time | relevance | path

Searched refs:TFE_ContextAsyncWait (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/c/eager/
Dc_api_distributed_test.cc166 TFE_ContextAsyncWait(ctx, status); in TestFunctionWithPackedInput()
379 TFE_ContextAsyncWait(ctx, status); in TEST()
502 TFE_ContextAsyncWait(ctx, status); in TestDistributedFunctionCancellation()
Dc_api_experimental.h369 TF_CAPI_EXPORT extern void TFE_ContextAsyncWait(TFE_Context* ctx,
Dc_api.cc242 TF_CAPI_EXPORT extern void TFE_ContextAsyncWait(TFE_Context* ctx, in TFE_ContextAsyncWait() function
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.cc312 TFE_ContextAsyncWait(context, await_status.get()); in Execute()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc756 TFE_ContextAsyncWait(tensorflow::InputTFE_Context(ctx), status.get()); in PYBIND11_MODULE()
762 TFE_ContextAsyncWait(tensorflow::InputTFE_Context(ctx), status.get()); in PYBIND11_MODULE()