Home
last modified time | relevance | path

Searched defs:k_blocks (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
H A Dq8gemm_sparse.cc447 auto k_blocks = (kc() + 4 - 1) / 4; variable
494 auto k_blocks = (kc() + 4 - 1) / 4; variable
568 auto k_blocks = (kc() + 4 - 1) / 4; variable
613 auto k_blocks = (kc() + 4 - 1) / 4; variable
/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_gemm_template.py288 def get_blocking(m_factor, n_factor, k_factor, m_blocks, n_blocks, k_blocks): argument
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dgemm-block-sparse-microkernel-tester.h380 auto k_blocks = (k() + kr_value - 1) / kr_value; in test_packed() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dgemm_interleaved_pretransposed_2d.hpp512 unsigned int k_blocks = iceildiv(args._Ksize, get_k_block_size(args)); in estimate_cycles() local
H A Dgemm_interleaved.hpp1146 unsigned int k_blocks = iceildiv(args._Ksize, get_k_block_size(args)); in estimate_cycles() local