HomeSort by relevance Sort by last modified time
    Searched refs:block_params (Results 1 - 5 of 5) sorted by null

  /external/gemmlowp/internal/
single_thread_gemm.h 71 BlockParams block_params; local
72 block_params.Init<KernelFormat>(rows, cols, depth, 1);
75 Side::Lhs, allocator, block_params);
77 Side::Rhs, allocator, block_params);
79 PackedResult packed_result(allocator, block_params);
83 const bool pack_rhs_once = block_params.l2_cols == cols;
89 for (int r = 0; r < rows; r += block_params.l2_rows) {
90 int rs = std::min(block_params.l2_rows, rows - r);
94 for (int c = 0; c < cols; c += block_params.l2_cols) {
95 int cs = std::min(block_params.l2_cols, cols - c)
    [all...]
block_params.h 15 // block_params.h: Logic to choose L1 and L2 block sizes
162 const BlockParams& block_params) {
164 side == Side::Lhs ? block_params.l1_rows : block_params.l1_cols;
166 side == Side::Lhs ? block_params.l2_rows : block_params.l2_cols;
168 side_block_params->l1_depth = block_params.l1_depth;
169 side_block_params->l2_depth = block_params.l2_depth;
multi_thread_gemm.h 423 BlockParams block_params; variable
424 block_params.Init<KernelFormat>(rows, cols, depth, 1);
426 PackedLhs packed_lhs(Side::Lhs, local_allocator, block_params);
428 PackedResult packed_result(local_allocator, block_params);
432 for (int c = 0; c < cols; c += block_params.l2_cols) {
433 int cs = std::min(block_params.l2_cols, cols - c);
435 for (int r = 0; r < rows; r += block_params.l2_rows) {
436 int rs = std::min(block_params.l2_rows, rows - r);
440 Compute(kernel, block_params, &packed_result, packed_lhs, packed_rhs);
607 BlockParams block_params; local
    [all...]
compute.h 22 #include "block_params.h"
90 void Compute(const KernelBase& kernel, const BlockParams& block_params,
95 kernel, block_params, packed_result, packed_lhs, packed_rhs);
pack.h 34 #include "block_params.h"
53 const BlockParams& block_params)
56 GetSideBlockParams(side, &params_, block_params);

Completed in 61 milliseconds