Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
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
/external/pytorch/torch/_inductor/codegen/
Dcpp_gemm_template.py288 def get_blocking(m_factor, n_factor, k_factor, m_blocks, n_blocks, k_blocks): argument
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
Dgemm-block-sparse-microkernel-tester.h380 auto k_blocks = (k() + kr_value - 1) / kr_value; in test_packed() local
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
Dgemm_interleaved_pretransposed_2d.hpp512 unsigned int k_blocks = iceildiv(args._Ksize, get_k_block_size(args)); in estimate_cycles() local
Dgemm_interleaved.hpp1146 unsigned int k_blocks = iceildiv(args._Ksize, get_k_block_size(args)); in estimate_cycles() local