Home
last modified time | relevance | path

Searched refs:block_params (Results 1 – 5 of 5) sorted by relevance

/external/gemmlowp/internal/
Dsingle_thread_gemm.h71 BlockParams block_params; in SingleThreadGemm() local
72 block_params.Init<KernelFormat>(rows, cols, depth, 1); in SingleThreadGemm()
75 Side::Lhs, allocator, block_params); in SingleThreadGemm()
77 Side::Rhs, allocator, block_params); in SingleThreadGemm()
79 PackedResult packed_result(allocator, block_params); in SingleThreadGemm()
83 const bool pack_rhs_once = block_params.l2_cols == cols; in SingleThreadGemm()
89 for (int r = 0; r < rows; r += block_params.l2_rows) { in SingleThreadGemm()
90 int rs = std::min(block_params.l2_rows, rows - r); in SingleThreadGemm()
94 for (int c = 0; c < cols; c += block_params.l2_cols) { in SingleThreadGemm()
95 int cs = std::min(block_params.l2_cols, cols - c); in SingleThreadGemm()
[all …]
Dblock_params.h162 const BlockParams& block_params) { in GetSideBlockParams() argument
164 side == Side::Lhs ? block_params.l1_rows : block_params.l1_cols; in GetSideBlockParams()
166 side == Side::Lhs ? block_params.l2_rows : block_params.l2_cols; in GetSideBlockParams()
168 side_block_params->l1_depth = block_params.l1_depth; in GetSideBlockParams()
169 side_block_params->l2_depth = block_params.l2_depth; in GetSideBlockParams()
Dmulti_thread_gemm.h423 BlockParams block_params; in Run() local
424 block_params.Init<KernelFormat>(rows, cols, depth, 1); in Run()
426 PackedLhs packed_lhs(Side::Lhs, local_allocator, block_params); in Run()
428 PackedResult packed_result(local_allocator, block_params); in Run()
432 for (int c = 0; c < cols; c += block_params.l2_cols) { in Run()
433 int cs = std::min(block_params.l2_cols, cols - c); in Run()
435 for (int r = 0; r < rows; r += block_params.l2_rows) { in Run()
436 int rs = std::min(block_params.l2_rows, rows - r); in Run()
440 Compute(kernel, block_params, &packed_result, packed_lhs, packed_rhs); in Run()
607 BlockParams block_params; in MultiThreadGemm() local
[all …]
Dcompute.h90 void Compute(const KernelBase& kernel, const BlockParams& block_params, in Compute() argument
95 kernel, block_params, packed_result, packed_lhs, packed_rhs); in Compute()
Dpack.h53 const BlockParams& block_params) in PackedSideBlock() argument
56 GetSideBlockParams(side, &params_, block_params); in PackedSideBlock()