1 #pragma once 2 #include <ATen/core/Tensor.h> 3 4 namespace at { namespace native { 5 6 Tensor& qembeddingbag_byte_prepack_out(Tensor& output, const Tensor& weight); 7 8 Tensor qembeddingbag_byte_prepack(const Tensor& weight); 9 10 Tensor qembeddingbag_byte_prepack_meta(const Tensor& weight); 11 12 } // namespace native 13 } // namespace at 14