treelite
Main Page
Modules
Classes
Files
File List
File Members
src
compiler
pred_transform.h
Go to the documentation of this file.
1
7
#ifndef TREELITE_PRED_TRANSFORM_H_
8
#define TREELITE_PRED_TRANSFORM_H_
9
10
#include <vector>
11
#include <string>
12
13
namespace
treelite
{
14
namespace
compiler {
15
16
inline
std::string PredTransformPrototype(
bool
batch) {
17
if
(batch) {
18
return
"size_t pred_transform_batch(float* pred, int64_t ndata, int nthread)"
;
19
}
else
{
20
return
"size_t pred_transform(float* pred)"
;
21
}
22
}
23
24
std::vector<std::string> PredTransformFunction(
const
Model& model,
bool
batch);
25
26
}
// namespace compiler
27
}
// namespace treelite
28
29
#endif // TREELITE_PRED_TRANSFORM_H_
treelite
Definition:
annotator.h:13
Generated by
1.8.11