Searched refs:get_cancellation_token (Results 1 – 16 of 16) sorted by relevance
28 auto token = manager->get_cancellation_token(); in TEST()41 auto token = manager->get_cancellation_token(); in TEST()52 auto token = manager->get_cancellation_token(); in TEST()62 auto token = manager->get_cancellation_token(); in TEST()76 auto token_1 = manager->get_cancellation_token(); in TEST()80 auto token_2 = manager->get_cancellation_token(); in TEST()90 auto token_3 = manager->get_cancellation_token(); in TEST()121 auto token = manager->get_cancellation_token(); in TEST()134 auto token = manager->get_cancellation_token(); in TEST()148 auto token = manager->get_cancellation_token(); in TEST()
64 CancellationToken get_cancellation_token();
56 CancellationToken token = cm->get_cancellation_token(); in Compute()
62 CancellationToken CancellationManager::get_cancellation_token() { in get_cancellation_token() function in tensorflow::CancellationManager
53 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()
74 CancellationToken token = cm->get_cancellation_token(); in TryEnqueue()139 CancellationToken token = cm->get_cancellation_token(); in TryEnqueueMany()200 CancellationToken token = cm->get_cancellation_token(); in TryDequeue()291 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
130 CancellationToken token = cm->get_cancellation_token(); in TryEnqueue()188 CancellationToken token = cm->get_cancellation_token(); in TryEnqueueMany()235 CancellationToken token = cm->get_cancellation_token(); in TryDequeue()327 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
73 CancellationToken token = cm->get_cancellation_token(); in TryTakeGrad()
117 token = cm->get_cancellation_token(); in AcquireAsync()
90 CancellationToken token = cm->get_cancellation_token(); in TryDequeueMany()
42 CancellationToken token = cancel_mgr_->get_cancellation_token(); in Start()
213 token = cancellation_manager_.get_cancellation_token(); in DoRunGraph()309 token = cancellation_manager_.get_cancellation_token(); in DoPartialRunGraph()
694 auto token = cm->get_cancellation_token(); in RunPartitionsHelper()
106 ? ctx->cancellation_manager()->get_cancellation_token() in CallContainer()
101 CancellationToken token = parent_mgr->get_cancellation_token(); in ComputeAsync()
588 cancellation_manager_->get_cancellation_token(); in RunInternal()