Searched defs:k_block (Results 1 – 8 of 8) sorted by relevance
| /external/XNNPACK/tools/ |
| D | generate-pack-test.py | 155 def generate_test_cases(ukernel, mr, k_block, isa): argument
|
| D | generate-spmm-test.py | 379 def generate_test_cases(ukernel, init_fn, mr, nr, k_block, is_pipelined, isa): argument
|
| D | generate-gemm-test.py | 875 def generate_test_cases(ukernel, mr, nr, kr, sr, xw, k_block, init_fn, argument
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| 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
|
| 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
|
| D | gemv_pretransposed.hpp | 96 unsigned int k_block=0; member in arm_gemm::GemvPretransposed
|
| 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
|
| D | gemm_interleaved_pretransposed_2d.hpp | 313 unsigned int k_block; in get_k_block_size() local
|