| /external/tensorflow/tensorflow/core/framework/ |
| D | cancellation_test.cc | 27 CancellationManager* manager = new CancellationManager(); in TEST()
40 CancellationManager* manager = new CancellationManager(); in TEST()
51 CancellationManager* manager = new CancellationManager(); in TEST()
61 CancellationManager* manager = new CancellationManager(); in TEST()
75 CancellationManager* manager = new CancellationManager(); in TEST()
99 CancellationManager* cm = new CancellationManager(); in TEST()
120 CancellationManager* manager = new CancellationManager(); in TEST()
133 CancellationManager* manager = new CancellationManager(); in TEST()
147 CancellationManager* manager = new CancellationManager(); in TEST()
|
| D | cancellation.cc | 23 const CancellationToken CancellationManager::kInvalidToken = -1;
25 CancellationManager::CancellationManager() in CancellationManager() function in tensorflow::CancellationManager
30 void CancellationManager::Reset() { in Reset()
36 void CancellationManager::StartCancel() { in StartCancel()
62 CancellationToken CancellationManager::get_cancellation_token() { in get_cancellation_token()
67 bool CancellationManager::RegisterCallback(CancellationToken token, in RegisterCallback()
78 bool CancellationManager::DeregisterCallback(CancellationToken token) { in DeregisterCallback()
98 bool CancellationManager::TryDeregisterCallback(CancellationToken token) { in TryDeregisterCallback()
108 CancellationManager::~CancellationManager() { in ~CancellationManager()
|
| D | cancellation.h | 45 class CancellationManager {
50 CancellationManager();
51 ~CancellationManager();
|
| D | collective.h | 29 class CancellationManager; variable
160 CancellationManager* cancel_mgr,
167 CancellationManager* cancel_mgr,
174 CancellationManager* cancel_mgr,
270 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 | 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 | collective_param_resolver_distributed.cc | 28 CancellationManager* cancel_mgr, in CompleteGroupCall()
51 bool is_source, CancellationManager* cancel_mgr, in CompleteInstanceCall()
96 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteParamsAsync()
114 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteGroupAsync()
146 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceAsync()
254 const string& device, CollectiveParams* cp, CancellationManager* cancel_mgr, in CompleteGroupDistributed()
364 CancellationManager* cancel_mgr, const StatusCallback& done) { in CompleteInstanceDistributed()
|
| D | cancellable_call.h | 30 CancellableCall(CancellationManager* cancel_mgr, const string& remote_worker, in CancellableCall()
57 CancellationManager* const cancel_mgr_; // Not owned
|
| D | partial_run_mgr.cc | 30 CancellationManager** cancellation_manager) { in FindOrCreate()
39 partial_run->cancellation_manager = MakeUnique<CancellationManager>(); in FindOrCreate()
|
| D | graph_mgr.h | 93 CancellationManager* cancellation_manager,
170 CancellationManager* cancellation_manager,
|
| D | worker.h | 28 class CancellationManager; variable
121 CancellationManager cancellation_manager_;
|
| D | collective_rma_distributed.h | 47 CancellationManager cancel_mgr_;
|
| /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 | 102 CleanUp(DoneCallback&& f, CancellationToken ct, CancellationManager* cm) in CleanUp()
106 CancellationManager* cm;
125 void Cancel(Action action, CancellationManager* cancellation_manager,
162 CancellationManager* cancellation_manager; // not owned
171 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()
|
| D | queue_base.cc | 188 void QueueBase::Cancel(Action action, CancellationManager* cancellation_manager, in Cancel()
249 0, callback, ctx, nullptr, CancellationManager::kInvalidToken, in Close()
327 if (to_clean.to_deregister != CancellationManager::kInvalidToken) { in FlushUnlocked()
|
| D | random_shuffle_queue_op.cc | 129 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueue()
187 CancellationManager* cm = ctx->cancellation_manager(); in TryEnqueueMany()
234 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeue()
326 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| /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/kernels/data/ |
| D | captured_function.cc | 320 CancellationManager c_mgr; in Run()
356 CancellationManager c_mgr; in RunWithBorrowedArgs()
392 CancellationManager c_mgr; in RunInstantiated()
436 CancellationManager* c_mgr = new CancellationManager(); in RunAsync()
|
| D | map_defun_op.cc | 94 CancellationManager* parent_mgr = ctx->cancellation_manager(); in ComputeAsync()
100 CancellationManager* c_mgr = new CancellationManager(); in ComputeAsync()
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_collective_executor_mgr.cc | 42 CancellationManager* cancel_mgr, WorkerCacheInterface* wc) in RecvBufCall()
144 CancellationManager cancel_mgr_;
|