Searched refs:my_ofm_start (Results 1 – 6 of 6) sorted by relevance
/dports/math/libxsmm/libxsmm-1.16.3/src/template/ |
H A D | libxsmm_dnn_convolve_st_upd_custom_custom_generic.tpl.c | 12 …2 = 0, oj, oi, ii, ij, kj, ki, ind, j_br, img_br, img_block_size = 1, my_ofm_start, my_ofm_end, my… variable 319 my_ofm_start = 0; 323 my_ofm_start = LIBXSMM_MIN(my_in_tile_id * ofms_per_thread, handle->blocksofm); 332 my_ofm_start = 0; 338 my_ofm_start = LIBXSMM_MIN(my_in_tile_id * ofms_per_thread, handle->blocksofm); 343 block_ofm = my_ofm_end-my_ofm_start+1; 434 for (ofm1 = my_ofm_start; ofm1 < my_ofm_end; ofm1++ ) { 451 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += block_ofm) { 485 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += block_ofm) {
|
H A D | libxsmm_dnn_convolve_st_fwd_custom_custom_generic.tpl.c | 19 int my_ofm_start = 0; variable 44 my_ofm_start = LIBXSMM_MIN(myOfmId * nOfmBlocks, handle->blocksofm); 67 my_ofm_start = LIBXSMM_MIN(ofm_id * ofmpt, handle->blocksofm); 276 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 358 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 417 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) {
|
H A D | libxsmm_dnn_convolve_st_fwd_nhwc_custom-rsck_generic.tpl.c | 19 int my_ofm_start = 0; variable 49 my_ofm_start = LIBXSMM_MIN(myOfmId * nOfmBlocks, handle->blocksofm); 72 my_ofm_start = LIBXSMM_MIN(ofm_id * ofmpt, handle->blocksofm); 302 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 399 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 463 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) {
|
H A D | libxsmm_dnn_convolve_st_fwd_custom_custom_generic_bf16.tpl.c | 20 int my_ofm_start = 0; variable 52 my_ofm_start = LIBXSMM_MIN(myOfmId * nOfmBlocks, handle->blocksofm); 75 my_ofm_start = LIBXSMM_MIN(ofm_id * ofmpt, handle->blocksofm); 296 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 421 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 493 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += handle->block_fwd_ofm) { 562 for (ofm1 = my_ofm_start; ofm1 < my_ofm_end; ofm1++) {
|
H A D | libxsmm_dnn_convolve_st_upd_custom_custom_generic_bf16.tpl.c | 67 …ofm2, oj, oi, ii, ij, kj, ki, j_br, img_br, i, j, img_block_size = 1, my_ofm_start, my_ofm_end, my… variable 470 my_ofm_start = 0; 474 my_ofm_start = LIBXSMM_MIN(my_in_tile_id * ofms_per_thread, handle->blocksofm); 483 my_ofm_start = 0; 488 block_ofm = my_ofm_end-my_ofm_start+1; 497 for (ofm1 = my_ofm_start; ofm1 < my_ofm_end; ofm1++ ) { 509 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += block_ofm) {
|
H A D | libxsmm_dnn_convolve_st_upd_nhwc_custom-rsck_generic.tpl.c | 12 …2 = 0, oj, oi, ii, ij, kj, ki, ind, j_br, img_br, img_block_size = 1, my_ofm_start, my_ofm_end, my… variable 388 my_ofm_start = 0; 392 my_ofm_start = LIBXSMM_MIN(my_in_tile_id * ofms_per_thread, handle->blocksofm); 401 my_ofm_start = 0; 406 block_ofm = my_ofm_end-my_ofm_start+1; 517 for (ofm1 = my_ofm_start; ofm1 < my_ofm_end; ofm1++ ) { 539 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += block_ofm) { 578 for (ofmb = my_ofm_start; ofmb < my_ofm_end; ofmb += block_ofm) {
|