File size: 386 Bytes
9dd3461 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
#pragma once
#ifdef USE_RUY_QMATMUL
#include <ruy/ruy.h>
namespace at {
namespace native {
namespace ruy_utils {
ruy::Context* get_ruy_context();
void quantize_multiplier(double scale,
int* multiplier_fixedpoint,
int* multiplier_exponent);
} // namespace ruy_utils
} // namespace native
} // namesplace
#endif // USE_RUY_QMATMUL
|