| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.h | 108 cancellation_manager_(cancellation_manager), in ExecuteNode()
125 graph_collector_, cancellation_manager_, retvals_, in Run()
143 CancellationManager* const cancellation_manager_; variable
164 cancellation_manager_(cancellation_manager), in AsyncExecuteNode()
204 cancellation_manager_, absl::MakeSpan(retvals_), stack_trace_); in Run()
238 CancellationManager* const cancellation_manager_; variable
|
| D | eager_operation.h | 178 return cancellation_manager_; in GetCancellationManager()
181 cancellation_manager_ = cancellation_manager; in SetCancellationManager()
253 CancellationManager* cancellation_manager_ = nullptr; // Not owned. variable
|
| D | eager_operation.cc | 345 cancellation_manager_ = nullptr; in Reset()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 212 token = cancellation_manager_.get_cancellation_token(); in DoRunGraph()
213 bool already_cancelled = !cancellation_manager_.RegisterCallback( in DoRunGraph()
235 cancellation_manager_.DeregisterCallback(token); in DoRunGraph()
314 token = cancellation_manager_.get_cancellation_token(); in DoPartialRunGraph()
315 cancellation_manager_.RegisterCallback(token, in DoPartialRunGraph()
321 cancellation_manager_.DeregisterCallback(token); in DoPartialRunGraph()
402 request, response, &cancellation_manager_, done); in CompleteGroupAsync()
415 request, response, &cancellation_manager_, done); in CompleteInstanceAsync()
|
| D | worker.h | 123 CancellationManager cancellation_manager_; variable
|
| D | master_session.h | 210 CancellationManager cancellation_manager_; variable
|
| D | master_session.cc | 1815 resp, &cancellation_manager_, is_last_partial_run); in DoPartialRun()
1987 &cancellation_manager_, false); in DoRunWithLocalExecution()
2057 &cancellation_manager_); in DoRunCallable()
2113 cancellation_manager_.StartCancel(); in Close()
2140 cancellation_manager_.StartCancel(); in GarbageCollect()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | model_dataset_op.cc | 128 cancellation_manager_ = absl::make_unique<CancellationManager>(); in Iterator()
132 ~Iterator() override { cancellation_manager_->StartCancel(); } in ~Iterator()
195 ram_budget_, cancellation_manager_.get()); in EnsureOptimizationLoopThreadStarted()
228 std::unique_ptr<CancellationManager> cancellation_manager_; member in tensorflow::data::ModelDatasetOp::Dataset::Iterator
|
| D | iterator_ops.h | 88 ~State() { cancellation_manager_.StartCancel(); } in ~State()
109 return &cancellation_manager_; in cancellation_manager()
120 CancellationManager cancellation_manager_; variable
|
| D | parallel_map_dataset_op.cc | 215 cancellation_manager_->StartCancel(); in ~Iterator()
226 cancellation_manager_ = in Initialize()
229 params.cancellation_manager = cancellation_manager_.get(); in Initialize()
649 std::unique_ptr<CancellationManager> cancellation_manager_; member in tensorflow::data::ParallelMapDatasetOp::Dataset::Iterator
|
| D | multi_device_iterator_ops.cc | 111 params.cancellation_manager = &cancellation_manager_; in GetNextFromShard()
147 CancellationManager* cancellation_manager() { return &cancellation_manager_; } in cancellation_manager()
385 CancellationManager cancellation_manager_; member in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
|
| D | dataset_test_base.cc | 357 params.cancellation_manager = cancellation_manager_.get(); in CreateIteratorContext()
494 cancellation_manager_ = absl::make_unique<CancellationManager>(); in CreateOpKernelContext()
495 params->cancellation_manager = cancellation_manager_.get(); in CreateOpKernelContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 53 cancellation_manager_(cancellation_manager), in RemoteExecuteNode()
131 CancellationManager* cancellation_manager_; variable
|
| D | remote_execute_node.cc | 52 CancellationManager* cm = cancellation_manager_; in RunAsync()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 131 CancellationManager* cancellation_manager_ TF_GUARDED_BY(execution_mutex_);
193 cancellation_manager_ = &cancellation_manager; in StartExecute()
213 cancellation_manager_ = nullptr; in Join()
244 TFE_OpSetCancellationManager(op_.get(), wrap(cancellation_manager_), status); in Execute()
248 cancellation_manager_->StartCancel(); in Execute()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | executor.cc | 366 CancellationManager* cancellation_manager_; member in tensorflow::__anonce8950920111::ExecutorState
412 cancellation_manager_(args.cancellation_manager), in ExecutorState()
704 params.cancellation_manager = cancellation_manager_; in Process()
1103 if (cancellation_manager_ && cancellation_manager_->IsCancelled() && in NodeDone()
1114 if (cancellation_manager_) { in NodeDone()
1126 if (cancellation_manager_) { in NodeDone()
1127 cancellation_manager_->StartCancel(); in NodeDone()
1276 if (cancellation_manager_) { in Finish()
1277 cancellation_manager_->StartCancel(); in Finish()
|
| D | permuter_test.cc | 376 op_params.cancellation_manager = &parent_->cancellation_manager_; in DoPermute()
443 CancellationManager cancellation_manager_; member in tensorflow::__anonfb3b1ff60111::PermuterTest
|
| D | direct_session.cc | 315 cancellation_manager_(new CancellationManager()), in DirectSession()
391 delete cancellation_manager_; in ~DirectSession()
685 CancellationManager step_cancellation_manager(cancellation_manager_); in RunInternal()
989 args.cancellation_manager = cancellation_manager_; in PRunSetup()
1112 cancellation_manager_, operation_timeout_in_ms_); in PRun()
1756 cancellation_manager_->StartCancel(); in Close()
|
| D | ring_gatherer_test.cc | 454 op_params.cancellation_manager = &parent_->cancellation_manager_; in DoGather()
536 CancellationManager cancellation_manager_; member in tensorflow::RingGathererTest
|
| D | ring_reducer_test.cc | 486 op_params.cancellation_manager = &parent_->cancellation_manager_; in DoReduce()
568 CancellationManager cancellation_manager_; member in tensorflow::RingReducerTest
|
| D | direct_session.h | 388 CancellationManager* cancellation_manager_; variable
|
| D | hierarchical_tree_broadcaster_test.cc | 634 op_params.cancellation_manager = &parent_->cancellation_manager_; in DoBroadcast()
727 CancellationManager cancellation_manager_; member in tensorflow::__anon04765bf90111::HierarchicalTreeBroadcasterTest
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.h | 123 CancellationManager cancellation_manager_; variable
|
| D | standalone.cc | 118 params.cancellation_manager = &cancellation_manager_; in MakeIterator()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | lookup_util.cc | 413 cancellation_manager_ = in Init()
415 params.cancellation_manager = cancellation_manager_.get(); in Init()
453 std::unique_ptr<CancellationManager> cancellation_manager_; member in tensorflow::lookup::DatasetIterator
|