Searched defs:row_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
22 row_block_size = 1 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
114 row_block_size, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/include/ |
| H A D | pack_block_sparse.h | 81 uint32_t row_block_size; // output features block size member
132 const uint32_t row_block_size, in generateBlockCSRMatrix()
209 const int64_t row_block_size, in generateBlockCSRMatrix()
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/sparse/quantized/dynamic/ |
| H A D | linear.py | 29 row_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 | 34 size_t row_block_size, in fillBlockSparseWeights()
|
| H A D | fully-connected-sparse-operator-tester.h | 32 size_t row_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 | 52 uint32_t row_block_size; member
|
| H A D | params.h | 585 uint32_t row_block_size; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/ |
| H A D | q8gemm_sparse.cc | 41 size_t row_block_size, in fillBlockSparseWeights()
|