Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/quantized/cpu/
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
/external/pytorch/aten/src/ATen/native/quantized/cuda/
DEmbeddingBag.cu173 const auto packed_weight_sizes = packed_weight.sizes(); in qembeddingbag_byte_unpack() local