Searched refs:SetCancelCallback (Results 1 – 12 of 12) sorted by relevance
231 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunGraphHandler()266 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RecvTensorHandlerRaw()296 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RecvBufHandler()314 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in CompleteGroupHandler()333 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in CompleteInstanceHandler()475 opts->SetCancelCallback( in GrpcRecvTensorAsync()
196 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunStepHandler()264 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunCallableHandler()
87 call_opts_->SetCancelCallback([this]() { context_->TryCancel(); }); in StartCall()
180 void SetCancelCallback(std::function<void()> callback) { in SetCancelCallback() function
27 opts.SetCancelCallback([&num_calls]() { num_calls++; }); in TEST()
60 void SetCancelCallback(CancelFunction cancel_func);
34 void CallOptions::SetCancelCallback(CancelFunction cancel_func) { in SetCancelCallback() function in tensorflow::CallOptions
208 opts->SetCancelCallback([this, cm, step_id]() { in DoRunGraph()300 opts->SetCancelCallback([this, cm, step_id]() { in DoPartialRunGraph()
87 opts->SetCancelCallback([this]() { in RecvBufAsync()
693 call_opts->SetCancelCallback([&calls]() { calls.StartCancel(); }); in RunPartitionsHelper()
76 opts->SetCancelCallback([this, step_id]() { AbortStep(step_id); }); in GrpcRecvTensorAsync()
54 context->SetCancelCallback( in CreateContextWithCancel()70 void SetCancelCallback(base::OnceClosure cancel_callback) { in SetCancelCallback() function in mojo::core::__anon696ef0640111::TriggerHelper::NotificationContext