Home
last modified time | relevance | path

Searched refs:scratch_di (Results 1 – 8 of 8) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_rnncell.c149 handle->scratch_di = 0; in libxsmm_dnn_create_rnncell()
1245 handle->scratch_di = (void*)address; in libxsmm_dnn_rnncell_bind_scratch()
1248 handle->scratch_di = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch()
1512 handle->scratch_di = (void*)address; in libxsmm_dnn_rnncell_bind_scratch()
1515 handle->scratch_di = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch()
1661 handle->scratch_di = 0; in libxsmm_dnn_rnncell_release_scratch()
1707 handle->scratch_di = 0; in libxsmm_dnn_rnncell_release_scratch()
H A Dlibxsmm_main.h705 void* scratch_di; member
/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck.tpl.c50 element_output_type *diD = (element_output_type*)handle->scratch_di;
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic.tpl.c51 element_output_type *diD = (element_output_type*)handle->scratch_di;
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic_bf16.tpl.c55 element_output_type *diD = (element_output_type*)handle->scratch_di;
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_bf16.tpl.c54 element_output_type *diD = (element_output_type*)handle->scratch_di;
H A Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c42 element_output_type *diD = (element_output_type*)handle->scratch_di;
H A Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c43 element_output_type *diD = (element_output_type*)handle->scratch_di;