Searched refs:scratch_diB (Results 1 – 8 of 8) sorted by relevance
/dports/math/libxsmm/libxsmm-1.16.3/src/ |
H A D | libxsmm_dnn_rnncell.c | 153 handle->scratch_diB = 0; in libxsmm_dnn_create_rnncell() 1277 handle->scratch_diB = (void*)address; in libxsmm_dnn_rnncell_bind_scratch() 1280 handle->scratch_diB = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch() 1544 handle->scratch_diB = (void*)address; in libxsmm_dnn_rnncell_bind_scratch() 1547 handle->scratch_diB = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch() 1665 handle->scratch_diB = 0; in libxsmm_dnn_rnncell_release_scratch() 1711 handle->scratch_diB = 0; in libxsmm_dnn_rnncell_release_scratch()
|
H A D | libxsmm_main.h | 709 void* scratch_diB; member
|
/dports/math/libxsmm/libxsmm-1.16.3/src/template/ |
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck.tpl.c | 157 LIBXSMM_VLA_DECL(4, element_output_type, diB, (element_output_type*)handle->scratch_diB, kBlocks, b…
|
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic.tpl.c | 170 LIBXSMM_VLA_DECL(4, element_output_type, diB, (element_output_type*)handle->scratch_diB, kBlocks, b…
|
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic_bf16.tpl.c | 159 LIBXSMM_VLA_DECL(5, element_output_type, diB, (element_output_type*)handle->scratch_diB, nBlocks, b…
|
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_bf16.tpl.c | 164 LIBXSMM_VLA_DECL(5, element_output_type, diB, (element_output_type*)handle->scratch_diB, nBlocks, b…
|
H A D | libxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c | 130 LIBXSMM_VLA_DECL(4, element_output_type, diB, (element_output_type*)handle->scratch_diB, kBlocks, …
|
H A D | libxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c | 145 LIBXSMM_VLA_DECL(4, element_output_type, diB, (element_output_type*)handle->scratch_diB, kBlocks, …
|