Home
last modified time | relevance | path

Searched refs:outer_block_size (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dredux_functor.h142 const Eigen::Index outer_block_size = Eigen::divup(outer_dim, num_blocks); in operator() local
159 const auto compute = [inner_dim, num_blocks, outer_block_size, in operator()
163 Eigen::Index outer_dim_start = start * outer_block_size; in operator()
164 Eigen::Index outer_dim_limit = limit * outer_block_size; in operator()
177 const Eigen::Index compute_size = outer_block_size * inner_dim; in operator()
251 const Eigen::Index outer_block_size = in operator() local
269 total_workload, num_blocks, outer_block_size, in operator()
273 Eigen::Index block_start = start * outer_block_size; in operator()
274 Eigen::Index block_limit = limit * outer_block_size; in operator()
307 const Eigen::Index compute_size = outer_block_size * inner_dim; in operator()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
H A Dgemv_pretransposed.hpp130 if (args._cfg && args._cfg->outer_block_size) {
131 n_block = args._cfg->outer_block_size;
243 c.outer_block_size = n_block; in get_config()
H A Dgemm_hybrid.hpp99 if (args._cfg && args._cfg->outer_block_size) { in compute_n_block()
100 unsigned int n_block = args._cfg->outer_block_size; in compute_n_block()
279 c.outer_block_size = _n_block; in get_config()
H A Dgemm_hybrid_quantized.hpp112 if (args._cfg && args._cfg->outer_block_size) { in compute_n_block()
113 unsigned int n_block = args._cfg->outer_block_size; in compute_n_block()
323 c.outer_block_size = _n_block; in get_config()
H A Dgemm_hybrid_quantized_inline.hpp111 if (args._cfg && args._cfg->outer_block_size) { in compute_n_block()
112 return args._cfg->outer_block_size; in compute_n_block()
H A Dgemm_hybrid_indirect.hpp340 if (args._cfg && args._cfg->outer_block_size) {
341 return args._cfg->outer_block_size;
783 c.outer_block_size = _n_block; in get_config()
H A Dgemm_interleaved_pretransposed_2d.hpp363 if (args._cfg && args._cfg->outer_block_size) { in GemmInterleavedPretransposed2d()
364 _x_block = args._cfg->outer_block_size; in GemmInterleavedPretransposed2d()
H A Dgemm_interleaved.hpp631 if (args._cfg && args._cfg->outer_block_size) { in get_x_block_size()
632 return roundup(args._cfg->outer_block_size, strategy::out_width()); in get_x_block_size()
1176 c.outer_block_size = _x_block; in get_config()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/assembly/
H A Darm_gemm.hpp111 unsigned int outer_block_size = 0; member