Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/eager/
Dc_api_experimental.h354 typedef struct TFE_CancellationManager TFE_CancellationManager; typedef
355 TF_CAPI_EXPORT extern TFE_CancellationManager* TFE_NewCancellationManager();
357 TFE_CancellationManager*);
359 TFE_CancellationManager*);
361 TFE_CancellationManager*);
366 typedef struct TFE_CancellationManager TFE_CancellationManager; typedef
368 TFE_Op* op, TFE_CancellationManager* cancellation_manager,
Dc_api_experimental.cc570 TFE_CancellationManager* TFE_NewCancellationManager() { in TFE_NewCancellationManager()
571 return new TFE_CancellationManager; in TFE_NewCancellationManager()
575 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerStartCancel()
580 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerIsCancelled()
585 TFE_CancellationManager* cancellation_manager) { in TFE_DeleteCancellationManager()
590 TFE_CancellationManager* cancellation_manager, in TFE_OpSetCancellationManager()
Dc_api_internal.h227 struct TFE_CancellationManager { struct
Dc_api_experimental_test.cc291 TFE_CancellationManager* c_mgr = TFE_NewCancellationManager(); in TEST()
/external/tensorflow/tensorflow/python/eager/
Dpywrap_tfe.h59 typedef struct TFE_CancellationManager TFE_CancellationManager; typedef
63 TFE_CancellationManager* cancellation_manager,
Dpywrap_tfe_src.cc839 TFE_CancellationManager* cancellation_manager, in TFE_Py_ExecuteCancelable()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc44 PYBIND11_MAKE_OPAQUE(TFE_CancellationManager);
217 TFE_CancellationManager* cancellation_manager, in InputTFE_OutputTensorHandles()
319 py::class_<TFE_CancellationManager> TFE_CancellationManager_class( in PYBIND11_MODULE()
579 TFE_CancellationManager& cancellation_manager, in PYBIND11_MODULE()