| /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_ GUARDED_BY(parent_->mu_) =
178 CancellationManager* next_sibling_ GUARDED_BY(parent_->mu_) =
|
| D | collective.h | 30 class CancellationManager; variable
173 CancellationManager* cancel_mgr,
180 CancellationManager* cancel_mgr,
187 CancellationManager* cancel_mgr,
286 CancellationManager* cancel_mgr, in CompleteParamsAsync()
|
| D | rendezvous_test.cc | 159 auto* cm = new CancellationManager(); in TEST_F()
173 auto* cm = new CancellationManager(); in TEST_F()
193 auto* cm = new CancellationManager(); in TEST_F()
213 auto* cm = new CancellationManager(); in TEST_F()
482 auto* cm = new CancellationManager(); in BM_PingPong()
|
| /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 | 35 CancellationManager* cancel_mgr,
40 CancellationManager* cancel_mgr,
45 CancellationManager* cancel_mgr,
63 CancellationManager* cancel_mgr,
81 CancellationManager* 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()
57 CancellationManager* const cancel_mgr_; // Not owned
|
| D | collective_param_resolver_distributed.cc | 30 CancellationManager* cancel_mgr, in CompleteGroupCall()
54 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall()
101 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync()
119 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
154 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
274 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteGroupDistributed()
370 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()
|
| D | worker.h | 29 class CancellationManager; variable
123 CancellationManager cancellation_manager_;
|
| D | graph_mgr.h | 94 CancellationManager* cancellation_manager,
171 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()
|
| D | fifo_queue.cc | 52 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueue()
110 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueueMany()
156 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeue()
247 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| D | priority_queue.cc | 73 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueue()
138 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueueMany()
199 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeue()
290 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 208 CancellationManager* cancellation_manager, in Run()
218 CancellationManager* cancellation_manager, in Run()
233 CancellationManager cancellation_manager;
239 std::vector<Tensor>* outputs, CancellationManager* cancellation_manager, in Run()
255 CancellationManager default_cancellation_manager; in Run()
308 std::vector<Tensor>* outputs, CancellationManager* cancellation_manager, in Run()
339 CancellationManager cm; in Run()
|
| D | kernel_and_device.h | 123 CancellationManager* cancellation_manager,
128 std::vector<Tensor>* outputs, CancellationManager* cancellation_manager,
189 CancellationManager* cancellation_manager,
195 CancellationManager* cancellation_manager,
269 CancellationManager* cancellation_manager,
274 CancellationManager* cancellation_manager,
|
| D | eager_operation.h | 99 CancellationManager* GetCancellationManager() const { in GetCancellationManager()
102 void SetCancellationManager(CancellationManager* cancellation_manager) { in SetCancellationManager()
145 CancellationManager* cancellation_manager_ = nullptr; // Not owned.
|
| D | execute_node.h | 79 CancellationManager* cancellation_manager, bool async, in ExecuteNode()
151 CancellationManager* const cancellation_manager_;
|
| /external/tensorflow/tensorflow/core/util/rpc/ |
| D | call_container.h | 107 : CancellationManager::kInvalidToken), in CallContainer()
115 if (token_ != CancellationManager::kInvalidToken) { in CallContainer()
133 if (token_ != CancellationManager::kInvalidToken) { in CallContainer()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.h | 48 CancellationManager* cancel_mgr,
53 CancellationManager* cancel_mgr,
58 CancellationManager* cancel_mgr,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.h | 148 CancellationManager* recv_cancellation() { return &recv_cancellation_; } in recv_cancellation()
152 CancellationManager recv_cancellation_;
|