def METHOD_NAME(self, old_proto=None):
    if not old_proto:
        old_proto = caffe2_pb2.NetDef()
    output = self._NNModule.METHOD_NAME(old_proto)
    new_proto = caffe2_pb2.NetDef()
    new_proto.ParseFromString(output)
    return new_proto