def pull_param(self, name):
pb = core_pb2.Tensor()
pb.name = name
res = self.pserver_stubs[0].pull_param(pb)
tensor = Tensor()
deserialize_from_pb(res, tensor)
return tensor
def pull_embedding_param(self, name, ids):
tensor = Tensor(name, None, ids)
pb = core_pb2.Tensor()
serialize_to_pb(tensor, pb)
res = self.pserver_stubs[0].pull_embedding_param(pb)
res_tensor = Tensor()
deserialize_from_pb(res, res_tensor)
return res_tensor
def push_grad(self, grad):
pb = core_pb2.Tensor()
serialize_to_pb(grad, pb)
self.pserver_stubs[0].push_grad(pb)
def push_param(self, param):
pb = core_pb2.Tensor()
serialize_to_pb(param, pb)
self.pserver_stubs[0].push_param(pb)