Searched defs:col_block_size (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/ |
| H A D | utils.py | 8 def _is_valid_linear_block_sparse_pattern(row_block_size, col_block_size): argument
23 col_block_size = 4 variable in LinearBlockSparsePattern
27 def __init__(self, row_block_size=1, col_block_size=4): argument
|
| H A D | linear.py | 19 def __init__(self, row_block_size=1, col_block_size=4, dtype=torch.qint8): argument
115 col_block_size, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/ |
| H A D | pack_block_sparse.h | 80 uint32_t col_block_size; // input features block size member
133 const uint32_t col_block_size, in generateBlockCSRMatrix()
210 const int64_t col_block_size) { in generateBlockCSRMatrix()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
| H A D | linear.py | 30 col_block_size, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/ |
| H A D | gemm-block-sparse-microkernel-tester.h | 35 size_t col_block_size, in fillBlockSparseWeights()
|
| H A D | fully-connected-sparse-operator-tester.h | 33 size_t col_block_size, in fillBlockSparseWeights()
|
| H A D | q8gemm_sparse.cc | 18 NR, row_block_size, col_block_size, \ argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| H A D | operator.h | 53 uint32_t col_block_size; member
|
| H A D | params.h | 586 uint32_t col_block_size; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/ |
| H A D | q8gemm_sparse.cc | 42 size_t col_block_size, in fillBlockSparseWeights()
|