23 lines
570 B
C++
23 lines
570 B
C++
#pragma once
|
|
#include <twml/common.h>
|
|
#include <twml/defines.h>
|
|
#include <twml/Tensor.h>
|
|
|
|
#ifdef __cplusplus
|
|
namespace twml {
|
|
TWMLAPI void discretizerInfer(
|
|
Tensor &output_keys,
|
|
Tensor &output_vals,
|
|
const Tensor &input_ids,
|
|
const Tensor &input_vals,
|
|
const Tensor &bin_ids,
|
|
const Tensor &bin_vals,
|
|
const Tensor &feature_offsets,
|
|
int output_bits,
|
|
const Map<int64_t, int64_t> &ID_to_index,
|
|
int start_compute,
|
|
int end_compute,
|
|
int output_start);
|
|
} // namespace twml
|
|
#endif
|