Home
last modified time | relevance | path

Searched defs:packed_weight_sizes (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag_unpack.cpp121 const auto packed_weight_sizes = packed_weight.sizes(); in qembeddingbag_byte_unpack_out() local
171 const auto packed_weight_sizes = packed_weight.sym_sizes(); in qembeddingbag_byte_unpack_meta() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DEmbeddingBag.cu173 const auto packed_weight_sizes = packed_weight.sizes(); in qembeddingbag_byte_unpack() local