Searched defs:k_block (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | gemm_hybrid_quantized_inline.hpp | 92 …unsigned int k_block = (L1_size / 2) / (sizeof(Toi) * (std::max(strategy::out_width(), strategy::o… in compute_k_block() local
115 const unsigned int k_block = compute_k_block(args); in compute_n_block() local
|
| H A D | gemm_hybrid_quantized.hpp | 93 …unsigned int k_block = (L1_size / 2) / (sizeof(Toi) * (std::max(strategy::out_width(), strategy::o… in compute_k_block() local
122 const unsigned int k_block = compute_k_block(args); in compute_n_block() local
|
| H A D | gemm_interleaved.hpp | 595 … unsigned int k_block = roundup(iceildiv(get_ktotal(args), num_k_blocks), strategy::k_unroll()); in get_k_block_size() local
601 unsigned int k_block; in get_k_block_size() local
637 const unsigned int k_block = get_k_block_size(args); in get_x_block_size() local
|
| H A D | gemv_pretransposed.hpp | 96 unsigned int k_block=0; member in arm_gemm::GemvPretransposed
|
| H A D | gemm_interleaved_pretransposed_2d.hpp | 313 unsigned int k_block; in get_k_block_size() local
|
| /aosp_15_r20/external/XNNPACK/tools/ |
| H A D | generate-pack-test.py | 155 def generate_test_cases(ukernel, mr, k_block, isa): argument
|
| H A D | generate-spmm-test.py | 379 def generate_test_cases(ukernel, init_fn, mr, nr, k_block, is_pipelined, isa): argument
|
| H A D | generate-gemm-test.py | 875 def generate_test_cases(ukernel, mr, nr, kr, sr, xw, k_block, init_fn, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/ |
| H A D | 8x4-packA-sse2.c | 59 for (uint32_t k_block = 0; k_block < num_k_blocks - 1; k_block++) { in pytorch_q8gemm_sparse_packA_ukernel_8x4__sse2() local
|