torch/include/ATen/native/quantized/cpu/qembeddingbag_prepack.h

#include <ATen/ATen.h>

namespace at {
namespace native {

Tensor& qembeddingbag_byte_prepack_out(Tensor& output, const Tensor& weight);

Tensor qembeddingbag_byte_prepack(const Tensor& weight);

} // namespace native
} // namespace at