Home
last modified time | relevance | path

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

/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_core_bf16.tpl.c168batchreduce_kerneld(&LIBXSMM_VLA_ACCESS(5, riT, ikb, KB*KB_BLOCKS, 0, 0, 0, kBlocks, bk_lp, bk, lp…
172batchreduce_kerneld(&LIBXSMM_VLA_ACCESS(5, rcT, ikb, KB*KB_BLOCKS, 0, 0, 0, kBlocks, bk_lp, bk, lp…
176batchreduce_kerneld(&LIBXSMM_VLA_ACCESS(5, rfT, ikb, KB*KB_BLOCKS, 0, 0, 0, kBlocks, bk_lp, bk, lp…
180batchreduce_kerneld(&LIBXSMM_VLA_ACCESS(5, roT, ikb, KB*KB_BLOCKS, 0, 0, 0, kBlocks, bk_lp, bk, lp…
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_core.tpl.c249 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
256 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
263 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
270 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
H A Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c123 const libxsmm_smmfunction_reducebatch_addr batchreduce_kerneld = libxsmm_smmdispatch_reducebatch_a… variable
362 batchreduce_kerneld(A_array, B_array, &LIBXSMM_VLA_ACCESS(2, dp, in, ik, K), &blocks);
450 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
457 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
H A Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c138 const libxsmm_smmfunction_reducebatch_addr batchreduce_kerneld = libxsmm_smmdispatch_reducebatch_a… variable
357 batchreduce_kerneld(A_array, B_array, &LIBXSMM_VLA_ACCESS(2, dp, in, ik, K), &blocks);
430 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
437 batchreduce_kerneld(A_array, B_array, dout_ptr, &blocks);
H A Dlibxsmm_dnn_rnncell_st_rnn_bwdupd_nc_kcck.tpl.c73 const libxsmm_smmfunction_reducebatch_addr batchreduce_kerneld = libxsmm_smmdispatch_reducebatch_ad… variable
289 batchreduce_kerneld(A_array, B_array, &LIBXSMM_VLA_ACCESS(3, delta, i, in, ik, N, K) , &blocks);
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck.tpl.c151 const libxsmm_smmfunction_reducebatch_addr batchreduce_kerneld = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic.tpl.c164 const libxsmm_smmfunction_reducebatch_addr batchreduce_kerneld = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic_bf16.tpl.c179 const libxsmm_bsmmfunction_reducebatch_strd batchreduce_kerneld = handle->bwdupd_kerneld; variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_bf16.tpl.c184 const libxsmm_bsmmfunction_reducebatch_strd batchreduce_kerneld = handle->bwdupd_kerneld; variable