def send_proto(self, iter_id, name, proto): any_proto = google.protobuf.any_pb2.Any() any_proto.Pack(proto) msg = tws_pb.TrainerWorkerMessage(data=tws_pb.DataMessage( iter_id=iter_id, name=name, any_data=any_proto)) self._transmit(msg) logging.debug('Data: send protobuf %s for iter %d. seq_num=%d.', name, iter_id, msg.seq_num)
def send(self, iter_id, name, x): msg = tws_pb.TrainerWorkerMessage(data=tws_pb.DataMessage( iter_id=iter_id, name=name, tensor=tf.make_tensor_proto(x))) self._transmit(msg) logging.debug('Data: send %s for iter %d. seq_num=%d.', name, iter_id, msg.seq_num)