Searched refs:cancellation_manager_ (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 213 token = cancellation_manager_.get_cancellation_token(); in DoRunGraph()
214 bool already_cancelled = !cancellation_manager_.RegisterCallback( in DoRunGraph()
234 cancellation_manager_.DeregisterCallback(token); in DoRunGraph()
309 token = cancellation_manager_.get_cancellation_token(); in DoPartialRunGraph()
310 cancellation_manager_.RegisterCallback(token, in DoPartialRunGraph()
316 cancellation_manager_.DeregisterCallback(token); in DoPartialRunGraph()
397 request, response, &cancellation_manager_, done); in CompleteGroupAsync()
410 request, response, &cancellation_manager_, done); in CompleteInstanceAsync()
|
| D | worker.h | 121 CancellationManager cancellation_manager_; variable
|
| D | master_session.h | 211 CancellationManager cancellation_manager_; variable
|
| D | master_session.cc | 1723 resp, &cancellation_manager_, is_last_partial_run); in DoPartialRun()
1888 &cancellation_manager_, false); in DoRunWithLocalExecution()
1958 &cancellation_manager_); in DoRunCallable()
2014 cancellation_manager_.StartCancel(); in Close()
2041 cancellation_manager_.StartCancel(); in GarbageCollect()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.cc | 280 cancellation_manager_(new CancellationManager()), in DirectSession()
348 delete cancellation_manager_; in ~DirectSession()
588 cancellation_manager_->get_cancellation_token(); in RunInternal()
589 const bool already_cancelled = !cancellation_manager_->RegisterCallback( in RunInternal()
662 if (!cancellation_manager_->DeregisterCallback(cancellation_token)) { in RunInternal()
883 args.cancellation_manager = cancellation_manager_; in PRunSetup()
1004 WaitForNotification(run_state, cancellation_manager_, in PRun()
1631 cancellation_manager_->StartCancel(); in Close()
|
| D | executor.cc | 1262 CancellationManager* cancellation_manager_; member in tensorflow::__anon675242740111::ExecutorState
1395 cancellation_manager_(args.cancellation_manager), in ExecutorState()
1633 params.cancellation_manager = cancellation_manager_; in Process()
2233 if (cancellation_manager_) { in NodeDone()
2234 cancellation_manager_->StartCancel(); in NodeDone()
2502 if (cancellation_manager_) { in Finish()
2503 cancellation_manager_->StartCancel(); in Finish()
|
| D | direct_session.h | 366 CancellationManager* cancellation_manager_; variable
|