Home
last modified time | relevance | path

Searched defs:batchreduce_kernela (Results 1 – 16 of 16) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tvm_cnnlayer/libxsmm_wrapper/
H A Dbatch_reduce_plus_init.cc17 …libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_addr(of… in batch_reduce_kernel_update() local
51 …libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_addr(of… in batch_reduce_kernel_init_update() local
/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_rnncell_st_rnn_fwd_nc_kcck.tpl.c42 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_gru_fwd_nc_kcck.tpl.c60 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_rnn_fwd_ncnc_kcck.tpl.c42 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_fwd_nc_kcck.tpl.c74 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_gru_fwd_nc_ck_generic.tpl.c75 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_rnn_bwdupd_nc_kcck.tpl.c74 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_fwd_nc_ck_generic.tpl.c92 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_fwd_nc_kcck_bf16.tpl.c135 const libxsmm_bsmmfunction_reducebatch_strd batchreduce_kernela = handle->fwd_kernela; variable
H A Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c116 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_a… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_fwd_nc_ck_generic_bf16.tpl.c151 const libxsmm_bsmmfunction_reducebatch_strd batchreduce_kernela = handle->fwd_kernela; variable
H A Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c131 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_a… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck.tpl.c146 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_bf16.tpl.c181 const libxsmm_bsmmfunction_reducebatch_strd batchreduce_kernela = handle->bwdupd_kernela; variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic.tpl.c159 const libxsmm_smmfunction_reducebatch_addr batchreduce_kernela = libxsmm_smmdispatch_reducebatch_ad… variable
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic_bf16.tpl.c176 const libxsmm_bsmmfunction_reducebatch_strd batchreduce_kernela = handle->bwdupd_kernela; variable