Home
last modified time | relevance | path

Searched refs:cancellation_token (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dbuf_rendezvous.cc31 h->cancellation_manager->DeregisterCallback(h->cancellation_token); in DeregisterCancellation()
33 h->cancellation_token = CancellationManager::kInvalidToken; in DeregisterCancellation()
65 h->cancellation_manager->TryDeregisterCallback(h->cancellation_token); in PurgeTable()
99 CancellationToken cancellation_token = CancellationManager::kInvalidToken; in ProvideBuf() local
103 cancellation_token = cancellation_manager->get_cancellation_token(); in ProvideBuf()
105 h = new Hook(cancellation_manager, cancellation_token); in ProvideBuf()
128 cancellation_token, [this, key]() { CancelHook(key); })) { in ProvideBuf()
191 CancellationToken cancellation_token = CancellationManager::kInvalidToken; in ConsumeBuf() local
194 cancellation_token = cancellation_manager->get_cancellation_token(); in ConsumeBuf()
196 cancellation_token, [this, key]() { CancelHook(key); }); in ConsumeBuf()
[all …]
Dbuf_rendezvous.h71 CancellationToken cancellation_token; member
73 CancellationToken cancellation_token) in Hook()
80 cancellation_token(cancellation_token) {} in Hook()
/external/tensorflow/tensorflow/core/kernels/
Dconditional_accumulator_base.h97 CancellationToken cancellation_token; member
103 CancellationToken cancellation_token, RunCallback run_callback) in Attempt()
108 cancellation_token(cancellation_token), in Attempt()
Dqueue_base.h159 CancellationToken cancellation_token; member
168 CancellationToken cancellation_token, RunCallback run_callback) in Attempt()
173 cancellation_token(cancellation_token), in Attempt()
Dconditional_accumulator_base.cc118 attempt.cancellation_token == token) { in Cancel()
155 cur_attempt->cancellation_token, in TryAttemptLocked()
Dqueue_base.cc198 attempt.cancellation_token == token) { in Cancel()
304 cur_attempt->cancellation_token, in TryAttemptLocked()
/external/tensorflow/tensorflow/core/framework/
Dlocal_rendezvous.cc43 CancellationToken cancellation_token) in Item()
46 recv_state.cancellation_token = cancellation_token; in Item()
75 CancellationToken cancellation_token; member
215 if (curr->recv_state.cancellation_token == token) { in RecvAsync()