FM / cpu

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/quantized/cpu/

/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/quantized/cpu

Name	Size	Mode	Actions
conv_packed_params.h	791	0644	edit dl rm
conv_serialization.h	11455	0644	edit dl rm
embedding_packed_params.h	888	0644	edit dl rm
fbgemm_utils.h	10887	0644	edit dl rm
init_qnnpack.h	146	0644	edit dl rm
packed_params.h	1467	0644	edit dl rm
qembeddingbag.h	895	0644	edit dl rm
qembeddingbag_prepack.h	236	0644	edit dl rm
qnnpack_utils.h	15853	0644	edit dl rm
quantized_ops.h	5860	0644	edit dl rm
quant_utils.h	7383	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/quantized/cpu/qembeddingbag.h (895B)

#include namespace at { namespace native { Tensor& embedding_bag_byte_rowwise_offsets_out( Tensor& output, const Tensor& weight, const Tensor& indices, const c10::optional& offsets_in, const bool /* scale_grad_by_freq */, const int64_t /* mode */, bool pruned_weights, const c10::optional& per_sample_weights_, const c10::optional& compressed_indices_mapping, bool include_last_offset); Tensor& embedding_bag_4bit_rowwise_offsets_out( Tensor& output, const Tensor& weight, const Tensor& indices, const c10::optional& offsets_in, const bool /* scale_grad_by_freq */, const int64_t /* mode */, bool pruned_weights, const c10::optional& per_sample_weights_, const c10::optional& compressed_indices_mapping, bool include_last_offset); } // native } // at