| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | partial_run_mgr_test.cc | 28 CancellationManager* cancellation_manager; in TEST() local
29 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager); in TEST()
30 EXPECT_TRUE(cancellation_manager != nullptr); in TEST()
37 CancellationManager* cancellation_manager; in TEST() local
38 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager); in TEST()
42 EXPECT_EQ(cancellation_manager, found_cancellation_manager); in TEST()
49 CancellationManager* cancellation_manager; in TEST() local
50 partial_run_mgr.FindOrCreate(step_id, &cancellation_manager); in TEST()
55 EXPECT_NE(cancellation_manager, new_cancellation_manager); in TEST()
63 CancellationManager* cancellation_manager; in TEST() local
[all …]
|
| D | partial_run_mgr.cc | 23 CancellationManager** cancellation_manager) { in FindOrCreate() argument
27 *cancellation_manager = it->second->cancellation_manager.get(); in FindOrCreate()
33 partial_run->cancellation_manager = in FindOrCreate()
35 *cancellation_manager = partial_run->cancellation_manager.get(); in FindOrCreate()
|
| D | partial_run_mgr.h | 53 bool FindOrCreate(int step_id, CancellationManager** cancellation_manager);
72 std::unique_ptr<CancellationManager> cancellation_manager; member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous.cc | 30 if (h->cancellation_manager != nullptr) { in DeregisterCancellation()
31 h->cancellation_manager->DeregisterCallback(h->cancellation_token); in DeregisterCancellation()
32 h->cancellation_manager = nullptr; in DeregisterCancellation()
64 if (h->cancellation_manager != nullptr) { in PurgeTable()
65 h->cancellation_manager->TryDeregisterCallback(h->cancellation_token); in PurgeTable()
90 CancellationManager* cancellation_manager) { in ProvideBuf() argument
102 if (cancellation_manager != nullptr) { in ProvideBuf()
103 cancellation_token = cancellation_manager->get_cancellation_token(); in ProvideBuf()
105 h = new Hook(cancellation_manager, cancellation_token); in ProvideBuf()
126 if (cancellation_manager != nullptr && in ProvideBuf()
[all …]
|
| D | buf_rendezvous.h | 70 CancellationManager* cancellation_manager; member
72 explicit Hook(CancellationManager* cancellation_manager, in Hook()
79 cancellation_manager(cancellation_manager), in Hook()
93 CancellationManager* cancellation_manager);
108 CancellationManager* cancellation_manager);
|
| D | collective_rma_local.cc | 31 CancellationManager* cancellation_manager, const StatusCallback& done) { in RecvFromPeer() argument
94 consumer_callback, cancellation_manager); in RecvFromPeer()
102 CancellationManager* cancellation_manager, const StatusCallback& done) { in PostToPeer() argument
106 from_alloc_attr, done, cancellation_manager); in PostToPeer()
|
| D | collective_rma_local.h | 46 CancellationManager* cancellation_manager,
55 CancellationManager* cancellation_manager,
|
| D | graph_runner.cc | 185 CancellationManager cancellation_manager; in Run() local
186 args.cancellation_manager = &cancellation_manager; in Run()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 238 CancellationManager cancellation_manager; member
245 CancellationManager* cancellation_manager, in Run() argument
263 if (cancellation_manager) { in Run()
264 params.cancellation_manager = cancellation_manager; in Run()
267 params.cancellation_manager = &op_execution_state->cancellation_manager; in Run()
275 params.cancellation_manager = &default_cancellation_manager; in Run()
324 CancellationManager* cancellation_manager, in Run() argument
329 RunAsync(step_container, inputs, outputs, cancellation_manager, in Run()
341 CancellationManager* cancellation_manager, in RunAsync() argument
377 if (cancellation_manager) { in RunAsync()
[all …]
|
| D | kernel_and_device.h | 134 CancellationManager* cancellation_manager,
149 CancellationManager* cancellation_manager,
211 CancellationManager* cancellation_manager,
218 CancellationManager* cancellation_manager, in RunAsync() argument
222 done(Run(step_container, inputs, outputs, cancellation_manager, in RunAsync()
300 CancellationManager* cancellation_manager,
307 CancellationManager* cancellation_manager,
|
| D | execute_node.h | 99 CancellationManager* cancellation_manager, in ExecuteNode() argument
108 cancellation_manager_(cancellation_manager), in ExecuteNode()
155 CancellationManager* cancellation_manager, in AsyncExecuteNode() argument
164 cancellation_manager_(cancellation_manager), in AsyncExecuteNode()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | cancellation.cc | 209 Status RegisterCancellationCallback(CancellationManager* cancellation_manager, in RegisterCancellationCallback() argument
212 if (cancellation_manager) { in RegisterCancellationCallback()
213 CancellationToken token = cancellation_manager->get_cancellation_token(); in RegisterCancellationCallback()
214 if (!cancellation_manager->RegisterCallback(token, std::move(callback))) { in RegisterCancellationCallback()
217 *deregister_fn = [cancellation_manager, token]() { in RegisterCancellationCallback()
218 cancellation_manager->DeregisterCallback(token); in RegisterCancellationCallback()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conditional_accumulator_base.cc | 72 CancellationManager* cm = ctx->cancellation_manager(); in TryTakeGrad()
111 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel() argument
117 if (attempt.cancellation_manager == cancellation_manager && in Cancel()
156 cur_attempt->context->cancellation_manager()); in TryAttemptLocked()
|
| D | conditional_accumulator_base.h | 96 CancellationManager* cancellation_manager; // not owned member
102 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
107 cancellation_manager(cancellation_manager), in Attempt()
138 void Cancel(CancellationManager* cancellation_manager,
|
| D | queue_base.h | 121 void Cancel(Action action, CancellationManager* cancellation_manager,
158 CancellationManager* cancellation_manager; // not owned member
167 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
172 cancellation_manager(cancellation_manager), in Attempt()
|
| D | queue_base.cc | 188 void QueueBase::Cancel(Action action, CancellationManager* cancellation_manager, in Cancel() argument
197 if (attempt.cancellation_manager == cancellation_manager && in Cancel()
305 cur_attempt->context->cancellation_manager()); in TryAttemptLocked()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 321 OpKernelContext* ctx, CancellationManager* cancellation_manager, in RegisterCancellation() argument
349 CancellationToken token = cancellation_manager->get_cancellation_token(); in RegisterCancellation()
352 bool already_cancelled = !cancellation_manager->RegisterCallback( in RegisterCancellation()
359 OpKernelContext* ctx, CancellationManager* cancellation_manager, in UnregisterCancellation() argument
397 cancellation_manager->TryDeregisterCallback(token); in UnregisterCancellation()
419 CancellationManager* cancellation_manager, OpKernelContext* ctx, in TPUExecute() argument
488 RegisterCancellation(ctx, cancellation_manager, device_ordinal); in TPUExecute()
509 already_cancelled = cancellation_manager->IsCancelling() || in TPUExecute()
510 cancellation_manager->IsCancelled(); in TPUExecute()
516 UnregisterCancellation(ctx, cancellation_manager, stream, device_ordinal, in TPUExecute()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | to_tf_record_op.cc | 82 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
83 params.cancellation_manager = &cancellation_manager; in DoCompute()
|
| D | io_ops.cc | 82 CancellationManager cancellation_manager(ctx->cancellation_manager()); in WriteData() local
83 params.cancellation_manager = &cancellation_manager; in WriteData()
99 if (ctx->cancellation_manager()->IsCancelled()) { in WriteData()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | iterator_ops.cc | 124 params.cancellation_manager = captured_state->cancellation_manager(); in GetNext()
127 ctx->cancellation_manager(), in GetNext()
128 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in GetNext()
212 params.cancellation_manager = new_state->cancellation_manager(); in Restore()
215 ctx->cancellation_manager(), in Restore()
216 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in Restore()
247 params.cancellation_manager = new_state->cancellation_manager(); in SetIteratorFromDataset()
250 ctx->cancellation_manager(), in SetIteratorFromDataset()
251 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in SetIteratorFromDataset()
640 CancellationManager cancellation_manager(ctx->cancellation_manager()); in DoCompute() local
[all …]
|
| D | captured_function.cc | 849 CancellationManager cancellation_manager(ctx->cancellation_manager()); in Run() local
850 f_opts.cancellation_manager = &cancellation_manager; in Run()
912 CancellationManager cancellation_manager(ctx->cancellation_manager()); in RunWithBorrowedArgs() local
913 f_opts.cancellation_manager = &cancellation_manager; in RunWithBorrowedArgs()
967 CancellationManager cancellation_manager; in RunInstantiated() local
968 f_opts.cancellation_manager = &cancellation_manager; in RunInstantiated()
1013 auto cancellation_manager = in RunAsync() local
1014 absl::make_unique<CancellationManager>(ctx->cancellation_manager()); in RunAsync()
1015 f_opts.cancellation_manager = cancellation_manager.get(); in RunAsync()
1027 cancellation_manager.release(); in RunAsync()
|
| D | multi_device_iterator_ops.cc | 111 params.cancellation_manager = &cancellation_manager_; in GetNextFromShard()
114 ctx->cancellation_manager(), in GetNextFromShard()
115 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in GetNextFromShard()
147 CancellationManager* cancellation_manager() { return &cancellation_manager_; } in cancellation_manager() function in tensorflow::data::__anon2e5a99280111::MultiDeviceIterator
567 params.cancellation_manager = resource->cancellation_manager(); in Compute()
571 ctx->cancellation_manager(), in Compute()
572 [cm = params.cancellation_manager]() { cm->StartCancel(); }, in Compute()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | transfer_ops.cc | 48 ctx->cancellation_manager()->get_cancellation_token(); in ComputeAsync()
54 already_cancelled = !ctx->cancellation_manager()->RegisterCallback( in ComputeAsync()
66 ctx->cancellation_manager()->DeregisterCallback(token); in ComputeAsync()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental.cc | 494 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerStartCancel() argument
495 tensorflow::unwrap(cancellation_manager)->StartCancel(); in TFE_CancellationManagerStartCancel()
499 TFE_CancellationManager* cancellation_manager) { in TFE_CancellationManagerIsCancelled() argument
500 return tensorflow::unwrap(cancellation_manager)->IsCancelled(); in TFE_CancellationManagerIsCancelled()
504 TFE_CancellationManager* cancellation_manager) { in TFE_DeleteCancellationManager() argument
505 delete tensorflow::unwrap(cancellation_manager); in TFE_DeleteCancellationManager()
509 TFE_CancellationManager* cancellation_manager, in TFE_OpSetCancellationManager() argument
513 operation->SetCancellationManager(tensorflow::unwrap(cancellation_manager)); in TFE_OpSetCancellationManager()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_execute_node.h | 43 CancellationManager* cancellation_manager, in RemoteExecuteNode() argument
53 cancellation_manager_(cancellation_manager), in RemoteExecuteNode()
|