Searched refs:GdrWorker (Results 1 - 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/contrib/gdr/
H A Dgdr_worker.h26 class GdrWorker : public GrpcWorker { class in namespace:tensorflow
28 GdrWorker(WorkerEnv* env, RemoteMemoryManager* remote_memory_manager);
H A Dgdr_worker.cc42 GdrWorker::GdrWorker(WorkerEnv* worker_env, function in class:tensorflow::GdrWorker
48 void GdrWorker::GrpcRecvTensorAsync(CallOptions* opts,
53 request->request_id(), "RecvTensor (GdrWorker)", *request);
H A Dgdr_server_lib.cc56 return std::unique_ptr<GdrWorker>(
57 new GdrWorker(env, remote_memory_manager_.get()));

Completed in 101 milliseconds