Searched defs:optimized_qparams (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag_prepack.cpp | 358 const bool optimized_qparams, in _qembeddingbag_nbit_prepack_helper()
509 const bool optimized_qparams, in qembeddingbag_4bit_prepack()
526 const bool optimized_qparams, in qembeddingbag_2bit_prepack()
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 4507 …def _test_embedding_bag_unpack_impl(self, pack_fn, unpack_fn, bit_rate, optimized_qparams, weights… argument
4549 optimized_qparams, num_batches, data_type=np.float32): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/ |
| H A D | insert_quant_dequant.cpp | 320 bool optimized_qparams = false; in insertEmbeddingBagOps() local
|