File size: 552 Bytes
1ce325b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#ifndef LM_INTERPOLATE_PIPELINE_H
#define LM_INTERPOLATE_PIPELINE_H

#include "../common/model_buffer.hh"
#include "../../util/fixed_array.hh"
#include "../../util/stream/config.hh"

#include <cstddef>
#include <string>

namespace lm { namespace interpolate {

struct Config {
  std::vector<float> lambdas;
  util::stream::SortConfig sort;
  std::size_t BufferSize() const { return sort.buffer_size; }
};

void Pipeline(util::FixedArray<ModelBuffer> &models, const Config &config, int write_file);

}} // namespaces
#endif // LM_INTERPOLATE_PIPELINE_H