Searched refs:SetCancelCallback (Results 1 – 11 of 11) sorted by relevance
248 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunGraphHandler()270 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RecvTensorHandlerRaw()289 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RecvBufHandler()307 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in CompleteGroupHandler()326 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in CompleteInstanceHandler()509 opts->SetCancelCallback( in GrpcRecvTensorAsync()
196 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunStepHandler()263 call->SetCancelCallback([call_opts]() { call_opts->StartCancel(); }); in RunCallableHandler()
192 void SetCancelCallback(std::function<void()> callback) { in SetCancelCallback() function
105 call_opts_->SetCancelCallback([this]() { context_->TryCancel(); }); in StartCall()
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
203 opts->SetCancelCallback([this, cm, step_id]() { in DoRunGraph()301 opts->SetCancelCallback([this, cm, step_id]() { in DoPartialRunGraph()
89 opts->SetCancelCallback([this]() { in RecvBufAsync()
744 call_opts->SetCancelCallback([&calls]() { in RunPartitionsHelper()
54 context->SetCancelCallback( in CreateContextWithCancel()70 void SetCancelCallback(base::OnceClosure cancel_callback) { in SetCancelCallback() function in mojo::core::__anone46d84e60111::TriggerHelper::NotificationContext