Home
last modified time | relevance | path

Searched refs:get_cancellation_token (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dcancellation_test.cc32 auto token = manager->get_cancellation_token(); in TEST()
45 auto token = manager->get_cancellation_token(); in TEST()
56 auto token = manager->get_cancellation_token(); in TEST()
66 auto token = manager->get_cancellation_token(); in TEST()
80 auto token_1 = manager->get_cancellation_token(); in TEST()
84 auto token_2 = manager->get_cancellation_token(); in TEST()
94 auto token_3 = manager->get_cancellation_token(); in TEST()
129 auto token = cm.get_cancellation_token(); in TEST()
150 auto token = manager->get_cancellation_token(); in TEST()
163 auto token = manager->get_cancellation_token(); in TEST()
[all …]
Dreader_op_kernel.h56 CancellationToken token = cm->get_cancellation_token(); in Compute()
Dcancellation.h71 CancellationToken get_cancellation_token() { in get_cancellation_token() function
Dcancellation.cc213 CancellationToken token = cancellation_manager->get_cancellation_token(); in RegisterCancellationCallback()
Dlocal_rendezvous.cc204 token = cm->get_cancellation_token(); in RecvAsync()
/external/tensorflow/tensorflow/core/kernels/
Dfifo_queue.cc53 CancellationToken token = cm->get_cancellation_token(); in TryEnqueue()
111 CancellationToken token = cm->get_cancellation_token(); in TryEnqueueMany()
157 CancellationToken token = cm->get_cancellation_token(); in TryDequeue()
248 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
Dpriority_queue.cc74 CancellationToken token = cm->get_cancellation_token(); in TryEnqueue()
139 CancellationToken token = cm->get_cancellation_token(); in TryEnqueueMany()
201 CancellationToken token = cm->get_cancellation_token(); in TryDequeue()
292 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
Drandom_shuffle_queue_op.cc131 CancellationToken token = cm->get_cancellation_token(); in TryEnqueue()
189 CancellationToken token = cm->get_cancellation_token(); in TryEnqueueMany()
236 CancellationToken token = cm->get_cancellation_token(); in TryDequeue()
328 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
Dconditional_accumulator_base.cc73 CancellationToken token = cm->get_cancellation_token(); in TryTakeGrad()
Dmutex_ops.cc85 token = cm->get_cancellation_token(); in AcquireAsync()
Dpadding_fifo_queue.cc90 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcancellable_call.cc24 CancellationToken token = cancel_mgr_->get_cancellation_token(); in Start()
Dworker.cc212 token = cancellation_manager_.get_cancellation_token(); in DoRunGraph()
314 token = cancellation_manager_.get_cancellation_token(); in DoPartialRunGraph()
Dcollective_param_resolver_distributed.cc301 abortion_cancel_mgr_.get_cancellation_token(); in CompleteGroupDistributed()
391 abortion_cancel_mgr_.get_cancellation_token(); in CompleteInstanceDistributed()
Dcollective_rma_distributed.cc176 abortion_cancel_mgr_.get_cancellation_token(); in RecvFromPeer()
Dbase_rendezvous_mgr.cc413 auto token = cm->get_cancellation_token(); in RegisterCall()
/external/tensorflow/tensorflow/core/common_runtime/
Dbuf_rendezvous.cc103 cancellation_token = cancellation_manager->get_cancellation_token(); in ProvideBuf()
194 cancellation_token = cancellation_manager->get_cancellation_token(); in ConsumeBuf()
Dcollective_param_resolver_local.cc104 const CancellationToken token = cancel_mgr->get_cancellation_token(); in CompleteGroupLocal()
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtransfer_ops.cc48 ctx->cancellation_manager()->get_cancellation_token(); in ComputeAsync()
Dtpu_compile_op_common.cc545 ctx->cancellation_manager()->get_cancellation_token(); in Compute()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Dremote_execute_node.cc56 token = cm->get_cancellation_token(); in RunAsync()
Dcluster_function_library_runtime.cc208 token = cm->get_cancellation_token(); in Run()
/external/tensorflow/tensorflow/core/nccl/
Dcollective_communicator.cc88 CancellationToken cancel_token = cancel_mgr->get_cancellation_token(); in Enqueue()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dtpu_compile_ops.cc126 ctx->cancellation_manager()->get_cancellation_token(); in Compute()
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc349 CancellationToken token = cancellation_manager->get_cancellation_token(); in RegisterCancellation()

12