Home
last modified time | relevance | path

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

/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_convolve_st_upd_custom_custom_generic_bf16.tpl.c269 n_blocks = handle->batchreduce_h_pixels;
274 for (oj = 0; oj < handle->ofh; oj += handle->batchreduce_h_pixels){
278 TRANS_OUTPUT_W_TO_VNNI_FORMAT(img, ofm1, oj, handle->batchreduce_h_pixels);
281 for (j=0; j < handle->batchreduce_h_pixels; j++) {
296 for (j_br = 0; j_br < handle->batchreduce_h_pixels; j_br++) {
304 … if ((oj + handle->batchreduce_h_pixels >= handle->ofh) && (img == my_img_end - 1)) {
336 n_blocks = handle->batchreduce_h_pixels;
346 for (oj = 0; oj < handle->ofh; oj += handle->batchreduce_h_pixels){
403 for (ij = 0; ij < handle->batchreduce_h_pixels; ij++) {
414 for (j_br = 0; j_br < handle->batchreduce_h_pixels; j_br++) {
[all …]
/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_main.h422 int batchreduce_h_pixels; member
H A Dlibxsmm_dnn_convolution.c799 … handle->batchreduce_h_pixels = (handle->upd_trans_w_only) ? 1 : 1; /* TODO: identical_branches */ in libxsmm_dnn_convolution_setup_bf16_upd()
800 handle->use_intermediate_f32_wt_tensor = (handle->batchreduce_h_pixels == handle->ofh) ? 0 : 1; in libxsmm_dnn_convolution_setup_bf16_upd()