MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / FuseRecvTensorHandlerRaw

Method FuseRecvTensorHandlerRaw

tensorflow/contrib/star/star_worker_service.cc:244–261  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

242}
243
244void StarWorkerService::FuseRecvTensorHandlerRaw(StarServerTag *tag) {
245 // LOG(INFO) << "StarWorkerService::FuseRecvTensorHandlerRaw";
246 Schedule([this, tag]() {
247 CallOptions* call_opts = new CallOptions;
248
249 StarCall<FuseRecvTensorRequest, StarFuseTensorResponse> *call =
250 new StarCall<FuseRecvTensorRequest, StarFuseTensorResponse>();
251
252 InitStarServerTag(&call->req_, &call->resp_, tag, [call] (const Status& s) {delete call;});
253
254 worker_->FuseRecvTensorAsync(call_opts, &call->req_, &call->resp_,
255 [tag, call, call_opts
256 ](const Status& s) {
257 delete call_opts;
258 tag->ProcessDone(s);
259 });
260 });
261}
262
263void StarWorkerService::Schedule(std::function<void()> f) {
264 worker_->env()->compute_pool->Schedule(std::move(f));

Callers

nothing calls this directly

Calls 3

InitStarServerTagFunction · 0.85
ProcessDoneMethod · 0.80
FuseRecvTensorAsyncMethod · 0.45

Tested by

no test coverage detected