def report_variable_to_master(self):
req = elasticdl_pb2.ReportVariableRequest()
for v in self._non_embed_vars.values():
emplace_tensor_pb_from_ndarray(req.variable,
v.numpy(),
name=v.name)
self._stub.ReportVariable(req)
def report_variable(self):
"""
report variable to ps.
"""
req = elasticdl_pb2.ReportVariableRequest()
for v in self._model.trainable_variables:
req.variable[v.name].CopyFrom(ndarray_to_tensor(v.numpy()))
self._stub.ReportVariable(req)
