Home
last modified time | relevance | path

Searched refs:cancellation_manager_ (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker.cc209 token = cancellation_manager_.get_cancellation_token(); in DoRunGraph()
210 bool already_cancelled = !cancellation_manager_.RegisterCallback( in DoRunGraph()
232 cancellation_manager_.DeregisterCallback(token); in DoRunGraph()
311 token = cancellation_manager_.get_cancellation_token(); in DoPartialRunGraph()
312 cancellation_manager_.RegisterCallback(token, in DoPartialRunGraph()
318 cancellation_manager_.DeregisterCallback(token); in DoPartialRunGraph()
399 request, response, &cancellation_manager_, done); in CompleteGroupAsync()
412 request, response, &cancellation_manager_, done); in CompleteInstanceAsync()
Dworker.h123 CancellationManager cancellation_manager_; variable
Dmaster_session.h209 CancellationManager cancellation_manager_; variable
Dmaster_session.cc1793 resp, &cancellation_manager_, is_last_partial_run); in DoPartialRun()
1965 &cancellation_manager_, false); in DoRunWithLocalExecution()
2035 &cancellation_manager_); in DoRunCallable()
2091 cancellation_manager_.StartCancel(); in Close()
2118 cancellation_manager_.StartCancel(); in GarbageCollect()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute_node.h87 cancellation_manager_(cancellation_manager), in ExecuteNode()
123 cancellation_manager_, absl::MakeSpan(retvals_)); in Run()
151 CancellationManager* const cancellation_manager_; variable
Deager_operation.h100 return cancellation_manager_; in GetCancellationManager()
103 cancellation_manager_ = cancellation_manager; in SetCancellationManager()
145 CancellationManager* cancellation_manager_ = nullptr; // Not owned. variable
Deager_operation.cc47 cancellation_manager_ = nullptr; in Reset()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc1330 CancellationManager* cancellation_manager_; member in tensorflow::__anonb108ab360111::ExecutorState
1468 cancellation_manager_(args.cancellation_manager), in ExecutorState()
1720 params.cancellation_manager = cancellation_manager_; in Process()
2289 if (cancellation_manager_ && cancellation_manager_->IsCancelled()) { in NodeDone()
2298 if (cancellation_manager_) { in NodeDone()
2313 if (cancellation_manager_) { in NodeDone()
2314 cancellation_manager_->StartCancel(); in NodeDone()
2585 if (cancellation_manager_) { in Finish()
2586 cancellation_manager_->StartCancel(); in Finish()
Ddirect_session.cc312 cancellation_manager_(new CancellationManager()), in DirectSession()
388 delete cancellation_manager_; in ~DirectSession()
671 CancellationManager step_cancellation_manager(cancellation_manager_); in RunInternal()
975 args.cancellation_manager = cancellation_manager_; in PRunSetup()
1098 cancellation_manager_, operation_timeout_in_ms_); in PRun()
1744 cancellation_manager_->StartCancel(); in Close()
Ddirect_session.h388 CancellationManager* cancellation_manager_; variable
/external/tensorflow/tensorflow/core/kernels/data/
Dmulti_device_iterator_ops.cc111 params.cancellation_manager = &cancellation_manager_; in GetNextFromShard()
147 CancellationManager* cancellation_manager() { return &cancellation_manager_; } in cancellation_manager()
383 CancellationManager cancellation_manager_; member in tensorflow::data::__anon1c56f74c0111::MultiDeviceIterator
Ddataset_test_base.cc350 params.cancellation_manager = cancellation_manager_.get(); in CreateIteratorContext()
485 cancellation_manager_ = absl::make_unique<CancellationManager>(); in CreateOpKernelContext()
486 params->cancellation_manager = cancellation_manager_.get(); in CreateOpKernelContext()
Ddataset_test_base.h727 std::unique_ptr<CancellationManager> cancellation_manager_; variable