Searched refs:fallback_to_no_sparse (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qembeddingbag.cpp | 242 bool fallback_to_no_sparse = false; in embedding_bag_nbit_impl() local
252 fallback_to_no_sparse = true; in embedding_bag_nbit_impl()
305 if (!pruned_weights || fallback_to_no_sparse) { in embedding_bag_nbit_impl()
378 (pruned_weights && !fallback_to_no_sparse)); in embedding_bag_nbit_impl()
391 (pruned_weights && !fallback_to_no_sparse)); in embedding_bag_nbit_impl()
414 bool fallback_to_no_sparse = false; in embedding_bag_byte_impl() local
424 fallback_to_no_sparse = true; in embedding_bag_byte_impl()
473 if (!pruned_weights || fallback_to_no_sparse) { in embedding_bag_byte_impl()
553 (pruned_weights && !fallback_to_no_sparse)); in embedding_bag_byte_impl()
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 4571 include_last_offset, fallback_to_no_sparse, sparsity, atol, rtol): argument
4633 if fallback_to_no_sparse:
4699 fallback_to_no_sparse=st.booleans(),
4707 fallback_to_no_sparse, argument
4713 fallback_to_no_sparse,
4724 fallback_to_no_sparse=st.booleans(),
4732 fallback_to_no_sparse, argument
4739 fallback_to_no_sparse,
4751 fallback_to_no_sparse=st.booleans(),
4759 fallback_to_no_sparse, argument
[all …]
|