Home
last modified time | relevance | path

Searched refs:cancellation_manager (Results 1 – 25 of 36) 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.h93 CancellationManager* cancellation_manager,
170 CancellationManager* cancellation_manager,
Dgraph_mgr.cc395 CancellationManager* cancellation_manager, in ExecuteAsync() argument
457 cancellation_manager, in ExecuteAsync()
473 CancellationManager* cancellation_manager, in StartParallelExecutors() argument
493 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()
138 void Cancel(CancellationManager* cancellation_manager,
Dconditional_accumulator_base.cc72 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()
Dqueue_base.h125 void Cancel(Action action, CancellationManager* cancellation_manager,
162 CancellationManager* cancellation_manager; // not owned member
171 OpKernelContext* context, CancellationManager* cancellation_manager, in Attempt()
176 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()
Dpartitioned_function_ops.cc217 run_opts.cancellation_manager = ctx->cancellation_manager(); in RunFunction()
Dfunction_ops.cc254 opts.cancellation_manager = ctx->cancellation_manager(); in ComputeAsync()
Dmutex_ops.cc112 CancellationManager* cm = c->cancellation_manager(); in AcquireAsync()
Dfunctional_ops.cc119 opts->cancellation_manager = ctx->cancellation_manager(); in SetRunOptions()
/external/tensorflow/tensorflow/core/util/rpc/
Dcall_container.h105 token_(ctx->cancellation_manager() != nullptr in CallContainer()
106 ? ctx->cancellation_manager()->get_cancellation_token() in CallContainer()
116 ctx_->cancellation_manager()->DeregisterCallback(token_); in CallContainer()
134 is_cancelled = !ctx_->cancellation_manager()->RegisterCallback( in CallContainer()
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_runner.cc184 CancellationManager cancellation_manager; in Run() local
185 args.cancellation_manager = &cancellation_manager; in Run()
Dexecutor.h89 CancellationManager* cancellation_manager = nullptr; member
Dfunction.cc485 opts.cancellation_manager = ctx->cancellation_manager(); in ComputeAsync()
938 exec_args->cancellation_manager = run_opts.cancellation_manager; in ExecutorArgsFromOptions()
1048 if (opts.cancellation_manager && opts.cancellation_manager->IsCancelled()) { in Run()
1118 if (opts.cancellation_manager && opts.cancellation_manager->IsCancelled()) { in Run()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc321 f_opts.cancellation_manager = &c_mgr; in Run()
357 f_opts.cancellation_manager = &c_mgr; in RunWithBorrowedArgs()
393 f_opts.cancellation_manager = &c_mgr; in RunInstantiated()
437 f_opts.cancellation_manager = c_mgr; in RunAsync()
Dmap_defun_op.cc94 CancellationManager* parent_mgr = ctx->cancellation_manager(); in ComputeAsync()
105 opts.cancellation_manager = c_mgr; in ComputeAsync()
Dsingle_threaded_executor.cc212 params.cancellation_manager = args.cancellation_manager; in RunAsync()
/external/tensorflow/tensorflow/core/framework/
Dreader_op_kernel.h55 CancellationManager* cm = context->cancellation_manager(); in Compute()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc242 params.cancellation_manager = &cm_; in Run()
320 opts.cancellation_manager = &cm_; in Run()

12