Searched defs:n_block (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | gemm_hybrid_quantized.hpp | 113 unsigned int n_block = args._cfg->outer_block_size; in compute_n_block() local
135 unsigned int n_block = (scaled_l2_size - k_block_area) / (sizeof(Toi) * k_block); in compute_n_block() local
|
| H A D | gemm_hybrid.hpp | 100 unsigned int n_block = args._cfg->outer_block_size; in compute_n_block() local
|
| H A D | gemm_hybrid_quantized_inline.hpp | 120 …unsigned int n_block = (((L2_size * 9) / 10) - (k_block * sizeof(Toi) * (strategy::out_width() + s… in compute_n_block() local
|
| H A D | gemv_pretransposed.hpp | 97 unsigned int n_block=0; member in arm_gemm::GemvPretransposed
|
| H A D | gemm_hybrid_indirect.hpp | 367 unsigned int n_block = iceildiv(args._Nsize, columns_needed); local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| H A D | flash_bwd_kernel.h | 80 …mpute_dq_dk_dv_1colblock(const Params ¶ms, const int bidb, const int bidh, const int n_block) { in compute_dq_dk_dv_1colblock()
805 for (int n_block = n_block_max - 2; n_block > 0; n_block--) { in compute_dq_dk_dv() local
823 …for (int n_block = blockIdx.x; n_block < (params.seqlen_k + Kernel_traits::kBlockN - 1) / Kernel_t… in compute_dq_dk_dv_seqk_parallel() local
|
| H A D | flash_bwd_preprocess_kernel.h | 145 const int n_block = blockIdx.x; in clear_dKVaccum() local
277 const int n_block = blockIdx.x; in convert_dKV() local
|
| H A D | flash_fwd_kernel.h | 249 int n_block = n_block_max - 1; in compute_attn_1rowblock() local
712 for (int n_block = n_block_max - 1; n_block >= n_block_copy_min; n_block--) { in compute_attn_1rowblock_splitkv() local
802 int n_block = n_block_max - 1; in compute_attn_1rowblock_splitkv() local
|
| /aosp_15_r20/packages/modules/Bluetooth/system/gd/crypto_toolbox/ |
| D | aes.cc | 603 return_type aes_cbc_encrypt(const unsigned char* in, unsigned char* out, int n_block,
652 return_type aes_cbc_decrypt(const unsigned char* in, unsigned char* out, int n_block,
|
| /aosp_15_r20/external/cronet/third_party/libxml/src/ |
| H A D | testapi.c | 37416 int n_block; in test_xmlUCSIsBlock() local
|