| /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()
|
| D | cancellation.cc | 23 const CancellationToken CancellationManager::kInvalidToken = -1;
25 CancellationManager::CancellationManager() in CancellationManager() function in tensorflow::CancellationManager
30 void CancellationManager::StartCancel() { in StartCancel()
56 CancellationToken CancellationManager::get_cancellation_token() { in get_cancellation_token()
61 bool CancellationManager::RegisterCallback(CancellationToken token, in RegisterCallback()
72 bool CancellationManager::DeregisterCallback(CancellationToken token) { in DeregisterCallback()
92 CancellationManager::~CancellationManager() { StartCancel(); } in ~CancellationManager()
|
| D | cancellation.h | 45 class CancellationManager {
50 CancellationManager();
51 ~CancellationManager();
|
| D | reader_op_kernel.h | 55 CancellationManager* cm = context->cancellation_manager(); in Compute()
|
| D | function.h | 35 class CancellationManager; variable
479 CancellationManager* cancellation_manager = nullptr;
|
| /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 | partial_run_mgr.h | 53 bool FindOrCreate(int step_id, CancellationManager** cancellation_manager);
72 std::unique_ptr<CancellationManager> cancellation_manager;
|
| D | partial_run_mgr.cc | 30 CancellationManager** cancellation_manager) { in FindOrCreate()
39 partial_run->cancellation_manager = MakeUnique<CancellationManager>(); in FindOrCreate()
|
| D | worker.cc | 29 : env_(env), cancellation_manager_(new CancellationManager) {} in Worker()
155 CancellationManager* cm = new CancellationManager; in DoRunGraph()
228 CancellationManager* cm = nullptr; in DoPartialRunGraph()
|
| D | worker.h | 28 class CancellationManager; variable
104 CancellationManager* cancellation_manager_ GUARDED_BY(mu_);
|
| D | graph_mgr.h | 92 CancellationManager* cancellation_manager,
166 CancellationManager* cancellation_manager,
|
| D | master_session.h | 192 CancellationManager cancellation_manager_;
|
| D | graph_mgr.cc | 385 CancellationManager* cancellation_manager, in ExecuteAsync()
451 CancellationManager* cancellation_manager, in StartParallelExecutors()
|
| /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;
137 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 | 67 CancellationManager* cm = ctx->cancellation_manager(); in TryTakeGrad()
106 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel()
175 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()
|
| D | padding_fifo_queue.cc | 89 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 209 CancellationManager c_mgr; in Run()
243 CancellationManager c_mgr; in RunWithBorrowedArgs()
290 auto c_mgr = new CancellationManager; in RunAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | direct_session.h | 251 void WaitForNotification(RunState* run_state, CancellationManager* cm,
314 CancellationManager* cancellation_manager_;
|
| D | executor.h | 88 CancellationManager* cancellation_manager = nullptr;
|
| D | direct_session.cc | 259 cancellation_manager_(new CancellationManager()), in DirectSession()
512 CancellationManager step_cancellation_manager; in Run()
1532 CancellationManager* cm, in WaitForNotification()
|