| /external/tensorflow/tensorflow/core/framework/ |
| D | cancellation.cc | 26 const CancellationToken CancellationManager::kInvalidToken = -1;
28 CancellationManager::CancellationManager() in CancellationManager() function in tensorflow::CancellationManager
33 CancellationManager::CancellationManager(CancellationManager* parent) in CancellationManager() function in tensorflow::CancellationManager
38 void CancellationManager::StartCancel() { in StartCancel()
40 std::forward_list<CancellationManager*> children_to_cancel; in StartCancel()
52 CancellationManager* child = state_->first_child; in StartCancel()
71 for (CancellationManager* child : children_to_cancel) { in StartCancel()
84 bool CancellationManager::RegisterCallback(CancellationToken token, in RegisterCallback()
98 bool CancellationManager::DeregisterCallback(CancellationToken token) { in DeregisterCallback()
124 bool CancellationManager::RegisterChild(CancellationManager* child) { in RegisterChild()
[all …]
|
| D | cancellation_test.cc | 31 CancellationManager* manager = new CancellationManager(); in TEST()
44 CancellationManager* manager = new CancellationManager(); in TEST()
55 CancellationManager* manager = new CancellationManager(); in TEST()
65 CancellationManager* manager = new CancellationManager(); in TEST()
79 CancellationManager* manager = new CancellationManager(); in TEST()
103 CancellationManager* cm = new CancellationManager(); in TEST()
124 CancellationManager cm; in TEST()
149 CancellationManager* manager = new CancellationManager(); in TEST()
162 CancellationManager* manager = new CancellationManager(); in TEST()
176 CancellationManager* manager = new CancellationManager(); in TEST()
[all …]
|
| D | cancellation.h | 48 class CancellationManager {
53 CancellationManager();
59 explicit CancellationManager(CancellationManager* parent);
61 ~CancellationManager();
156 CancellationManager* first_child = nullptr; // Not owned.
159 bool RegisterChild(CancellationManager* child);
160 void DeregisterChild(CancellationManager* child);
166 CancellationManager* const parent_ = nullptr; // Not owned.
176 CancellationManager* prev_sibling_ TF_GUARDED_BY(parent_->mu_) =
178 CancellationManager* next_sibling_ TF_GUARDED_BY(parent_->mu_) =
[all …]
|
| D | collective.h | 30 class CancellationManager; variable
182 CancellationManager* cancel_mgr,
189 CancellationManager* cancel_mgr,
196 CancellationManager* cancel_mgr,
271 CancellationManager* cancellation_manager,
280 CancellationManager* cancellation_manager,
310 CancellationManager* cancel_mgr, in CompleteParamsAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | partial_run_mgr_test.cc | 28 CancellationManager* cancellation_manager; in TEST()
37 CancellationManager* cancellation_manager; in TEST()
40 CancellationManager* found_cancellation_manager; in TEST()
49 CancellationManager* cancellation_manager; in TEST()
53 CancellationManager* new_cancellation_manager; in TEST()
63 CancellationManager* cancellation_manager; in TEST()
111 CancellationManager* cancellation_manager; in TEST_P()
127 CancellationManager* cancellation_manager; in TEST_P()
|
| D | collective_param_resolver_distributed.h | 38 CancellationManager* cancel_mgr,
43 CancellationManager* cancel_mgr,
48 CancellationManager* cancel_mgr,
69 CancellationManager* cancel_mgr,
87 CancellationManager* cancel_mgr,
93 CancellationManager abortion_cancel_mgr_;
|
| D | partial_run_mgr.h | 53 bool FindOrCreate(int step_id, CancellationManager** cancellation_manager);
72 std::unique_ptr<CancellationManager> cancellation_manager;
|
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall()
53 CancellationManager* const cancel_mgr_; // Not owned
|
| D | collective_rma_distributed.h | 46 CancellationManager* cancellation_manager,
59 CancellationManager abortion_cancel_mgr_;
|
| D | collective_param_resolver_distributed.cc | 37 CancellationManager* cancel_mgr, in CompleteGroupCall()
61 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall()
109 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteParamsAsync()
133 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
169 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
288 CancellationManager* cancel_mgr, const GroupRecCallback& done) { in CompleteGroupDistributed()
380 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceDistributed()
|
| D | partial_run_mgr.cc | 23 CancellationManager** cancellation_manager) { in FindOrCreate()
34 tensorflow::MakeUnique<CancellationManager>(); in FindOrCreate()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.h | 70 CancellationManager* cancellation_manager;
72 explicit Hook(CancellationManager* cancellation_manager, in Hook()
93 CancellationManager* cancellation_manager);
108 CancellationManager* cancellation_manager);
|
| D | collective_param_resolver_local.h | 51 CancellationManager* cancel_mgr,
56 CancellationManager* cancel_mgr,
61 CancellationManager* cancel_mgr,
89 CancellationManager* cancel_mgr)
|
| D | buf_rendezvous.cc | 33 h->cancellation_token = CancellationManager::kInvalidToken; in DeregisterCancellation()
90 CancellationManager* cancellation_manager) { in ProvideBuf()
99 CancellationToken cancellation_token = CancellationManager::kInvalidToken; in ProvideBuf()
152 CancellationManager* cancellation_manager) { in ConsumeBuf()
191 CancellationToken cancellation_token = CancellationManager::kInvalidToken; in ConsumeBuf()
|
| D | test_collective_executor_mgr.h | 40 CancellationManager* cancel_mgr, in CompleteParamsAsync()
47 CancellationManager* cancel_mgr, in CompleteGroupAsync()
54 CancellationManager* cancel_mgr, in CompleteInstanceAsync()
|
| D | collective_rma_local.h | 46 CancellationManager* cancellation_manager,
55 CancellationManager* cancellation_manager,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base.h | 96 CancellationManager* cancellation_manager; // not owned
102 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
116 CleanUp(DoneCallback&& f, CancellationToken ct, CancellationManager* cm) in CleanUp()
120 CancellationManager* cm;
138 void Cancel(CancellationManager* cancellation_manager,
|
| D | queue_base.h | 98 CleanUp(DoneCallback&& f, CancellationToken ct, CancellationManager* cm) in CleanUp()
102 CancellationManager* cm;
121 void Cancel(Action action, CancellationManager* cancellation_manager,
158 CancellationManager* cancellation_manager; // not owned
167 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
|
| D | conditional_accumulator_base.cc | 72 CancellationManager* cm = ctx->cancellation_manager(); in TryTakeGrad()
111 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel()
180 if (to_clean.to_deregister != CancellationManager::kInvalidToken) { in FlushUnlocked()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tfe_cancellation_manager_internal.h | 25 DEFINE_CONVERSION_FUNCTIONS(tensorflow::CancellationManager,
27 DEFINE_CONVERSION_FUNCTIONS(tensorflow::CancellationManager*,
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 134 CancellationManager* cancellation_manager,
149 CancellationManager* cancellation_manager,
211 CancellationManager* cancellation_manager,
218 CancellationManager* cancellation_manager, in RunAsync()
300 CancellationManager* cancellation_manager,
307 CancellationManager* cancellation_manager,
|
| D | kernel_and_device.cc | 238 CancellationManager cancellation_manager;
245 CancellationManager* cancellation_manager, in Run()
262 CancellationManager default_cancellation_manager; in Run()
324 CancellationManager* cancellation_manager, in Run()
341 CancellationManager* cancellation_manager, in RunAsync()
376 std::shared_ptr<CancellationManager> local_cm; in RunAsync()
380 local_cm = std::make_shared<CancellationManager>(); in RunAsync()
|
| D | execute_node.h | 99 CancellationManager* cancellation_manager, in ExecuteNode()
143 CancellationManager* const cancellation_manager_;
155 CancellationManager* cancellation_manager, in AsyncExecuteNode()
238 CancellationManager* const cancellation_manager_;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.h | 151 CancellationManager* recv_cancellation() { return &recv_cancellation_; } in recv_cancellation()
155 CancellationManager recv_cancellation_;
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.cc | 91 CancellationManager& cancellation_manager);
131 CancellationManager* cancellation_manager_ TF_GUARDED_BY(execution_mutex_);
180 CancellationManager& cancellation_manager) { in StartExecute()
261 default_cancellation_manager_(absl::make_unique<CancellationManager>()) { in ParallelDevice()
315 default_cancellation_manager_ = absl::make_unique<CancellationManager>(); in Execute()
323 int expected_max_outputs, CancellationManager& cancellation_manager) const { in StartExecute()
|