Home
last modified time | relevance | path

Searched refs:cancellation_manager (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/distributed_runtime/
Dpartial_run_mgr_test.cc28 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 …]
Dpartial_run_mgr.cc30 CancellationManager** cancellation_manager) { in FindOrCreate() argument
34 *cancellation_manager = it->second->cancellation_manager.get(); in FindOrCreate()
39 partial_run->cancellation_manager = MakeUnique<CancellationManager>(); in FindOrCreate()
40 *cancellation_manager = partial_run->cancellation_manager.get(); in FindOrCreate()
Dpartial_run_mgr.h53 bool FindOrCreate(int step_id, CancellationManager** cancellation_manager);
72 std::unique_ptr<CancellationManager> cancellation_manager; member
Dgraph_mgr.h92 CancellationManager* cancellation_manager,
166 CancellationManager* cancellation_manager,
Dgraph_mgr.cc385 CancellationManager* cancellation_manager, in ExecuteAsync() argument
439 cost_graph, cancellation_manager, in ExecuteAsync()
451 CancellationManager* cancellation_manager, in StartParallelExecutors() argument
473 args.cancellation_manager = cancellation_manager; in StartParallelExecutors()
/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator_base.h96 CancellationManager* cancellation_manager; // not owned member
102 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
107 cancellation_manager(cancellation_manager), in Attempt()
137 void Cancel(CancellationManager* cancellation_manager,
Dconditional_accumulator_base.cc67 CancellationManager* cm = ctx->cancellation_manager(); in TryTakeGrad()
106 CancellationManager* cancellation_manager, CancellationToken token) { in Cancel() argument
112 if (attempt.cancellation_manager == cancellation_manager && in Cancel()
151 cur_attempt->context->cancellation_manager()); in TryAttemptLocked()
Dqueue_base.h121 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()
Dqueue_base.cc188 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()
Dfifo_queue.cc52 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()
Dpriority_queue.cc73 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()
Drandom_shuffle_queue_op.cc129 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()
Dcritical_section.cc188 opts_.cancellation_manager = ctx_->cancellation_manager(); in ExecuteFunction()
Dfunctional_ops.cc102 opts->cancellation_manager = ctx->cancellation_manager(); in SetRunOptions()
Dfunction_ops.cc246 opts.cancellation_manager = ctx->cancellation_manager(); in ComputeAsync()
Dpadding_fifo_queue.cc89 CancellationManager* cm = ctx->cancellation_manager(); in TryDequeueMany()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc210 f_opts.cancellation_manager = &c_mgr; in Run()
244 f_opts.cancellation_manager = &c_mgr; in RunWithBorrowedArgs()
291 f_opts.cancellation_manager = c_mgr; in RunAsync()
Diterator_ops.cc769 opts.cancellation_manager = ctx->cancellation_manager(); in TryInit()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc300 opts.cancellation_manager = ctx->cancellation_manager(); in ComputeAsync()
754 if (opts.cancellation_manager && opts.cancellation_manager->IsCancelled()) { in Run()
780 exec_args->cancellation_manager = run_opts.cancellation_manager; in Run()
828 if (opts.cancellation_manager && opts.cancellation_manager->IsCancelled()) { in Run()
866 exec_args->cancellation_manager = run_opts.cancellation_manager; in Run()
Dexecutor.h88 CancellationManager* cancellation_manager = nullptr; member
Ddirect_session.cc527 args.cancellation_manager = &step_cancellation_manager; in Run()
767 args.cancellation_manager = cancellation_manager_; in PRunSetup()
Dexecutor.cc1305 cancellation_manager_(args.cancellation_manager), in ExecutorState()
1510 params.cancellation_manager = cancellation_manager_; in Process()
/external/tensorflow/tensorflow/core/framework/
Dreader_op_kernel.h55 CancellationManager* cm = context->cancellation_manager(); in Compute()
Dop_kernel.h543 CancellationManager* cancellation_manager = nullptr; member
1005 CancellationManager* cancellation_manager() const { in cancellation_manager() function
1006 return params_->cancellation_manager; in cancellation_manager()
Dfunction.h479 CancellationManager* cancellation_manager = nullptr; member

12