Home
last modified time | relevance | path

Searched refs:_test_embedding_bag_unpack_impl (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_op.py4507 …def _test_embedding_bag_unpack_impl(self, pack_fn, unpack_fn, bit_rate, optimized_qparams, weights… member in TestQuantizedEmbeddingOps
4556 …self._test_embedding_bag_unpack_impl(pack_fn, unpack_fn, bit_rate, optimized_qparams, unsplit_weig…
4562 … self._test_embedding_bag_unpack_impl(pack_fn, unpack_fn, bit_rate, optimized_qparams, weight)