mirror of
https://github.com/twitter/the-algorithm.git
synced 2024-12-23 10:41:50 +01:00
ef4c5eb65e
Please note we have force-pushed a new initial commit in order to remove some publicly-available Twitter user information. Note that this process may be required in the future.
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
|