Searched defs:pack_fn (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/test/quantization/core/ |
| D | test_quantized_op.py | 4507 …def _test_embedding_bag_unpack_impl(self, pack_fn, unpack_fn, bit_rate, optimized_qparams, weights… argument
4548 …def _test_embedding_bag_unpack_fn(self, pack_fn, unpack_fn, num_embeddings, embedding_dim, bit_rat… argument
|