Home
last modified time | relevance | path

Searched defs:num_k_blocks (Results 1 – 3 of 3) sorted by relevance

/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/
D8x4-packA-sse2.c58 uint32_t num_k_blocks = (K + COL_BLOCK_SIZE -1) / COL_BLOCK_SIZE; in pytorch_q8gemm_sparse_packA_ukernel_8x4__sse2() local
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
Dgemm_interleaved.hpp593 unsigned int num_k_blocks = iceildiv(get_ktotal(args), max_block_size); in get_k_block_size() local
612 unsigned int num_k_blocks = iceildiv(get_ktotal(args), k_block); in get_k_block_size() local
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1566 const int num_k_blocks = (right_dim0 + KR - 1) / KR; in Compute() local