Home
last modified time | relevance | path

Searched refs:nr_block_size (Results 1 – 3 of 3) sorted by relevance

/external/XNNPACK/src/xnnpack/
Dpack.h31 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_gemm_goi_w() local
34 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_q8_gemm_goi_w()
39 size_t n = nr_block_size; in xnn_pack_q8_gemm_goi_w()
45 packed_w = (void*) ((uintptr_t) packed_w + (nr - nr_block_size) * sizeof(int32_t)); in xnn_pack_q8_gemm_goi_w()
48 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_q8_gemm_goi_w()
59 packed_w = (void*) ((uintptr_t) packed_w + (nr - nr_block_size) * kr * sizeof(uint8_t)); in xnn_pack_q8_gemm_goi_w()
82 const size_t nr_block_size = min(nc - nr_block_start, nr); in xnn_pack_q8_gemm_io_w() local
85 for (size_t nr_block_offset = 0; nr_block_offset < nr_block_size; nr_block_offset++) { in xnn_pack_q8_gemm_io_w()
90 size_t n = nr_block_size; in xnn_pack_q8_gemm_io_w()
96 packed_w = (void*) ((uintptr_t) packed_w + (nr - nr_block_size) * sizeof(int32_t)); in xnn_pack_q8_gemm_io_w()
[all …]
Dcompute.h74 size_t nr_block_size);
81 size_t nr_block_size);
154 size_t nr_block_size);
162 size_t nr_block_size);
196 size_t nr_block_size);
206 size_t nr_block_size);
/external/XNNPACK/src/
Doperator-run.c29 size_t nr_block_size) in xnn_compute_ggemm() argument
37 nr_block_size, in xnn_compute_ggemm()
53 size_t nr_block_size) in xnn_compute_gemm() argument
60 nr_block_size, in xnn_compute_gemm()
95 size_t nr_block_size) in xnn_compute_gigemm() argument
102 nr_block_size, in xnn_compute_gigemm()
121 size_t nr_block_size) in xnn_compute_igemm() argument
128 nr_block_size, in xnn_compute_igemm()