Searched refs:scratch_wT (Results 1 – 10 of 10) sorted by relevance
/dports/math/libxsmm/libxsmm-1.16.3/src/template/ |
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck.tpl.c | 57 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; 82 element_filter_type *scratch_wiT = &(scratch_wT[0]); 83 element_filter_type *scratch_wcT = &(scratch_wT[C*K]); 84 element_filter_type *scratch_wfT = &(scratch_wT[2*C*K]); 85 element_filter_type *scratch_woT = &(scratch_wT[3*C*K]);
|
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic.tpl.c | 57 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; variable 90 element_filter_type *scratch_wiT = &(scratch_wT[0]); 91 element_filter_type *scratch_wcT = &(scratch_wT[C*K]); 92 element_filter_type *scratch_wfT = &(scratch_wT[2*C*K]); 93 element_filter_type *scratch_woT = &(scratch_wT[3*C*K]);
|
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic_bf16.tpl.c | 64 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; variable 101 element_filter_type *scratch_wiT = &(scratch_wT[0]); 102 element_filter_type *scratch_wcT = &(scratch_wT[C*K]); 103 element_filter_type *scratch_wfT = &(scratch_wT[2*C*K]); 104 element_filter_type *scratch_woT = &(scratch_wT[3*C*K]);
|
H A D | libxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_bf16.tpl.c | 64 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; 103 element_filter_type *scratch_wiT = &(scratch_wT[0]); 104 element_filter_type *scratch_wcT = &(scratch_wT[C*K]); 105 element_filter_type *scratch_wfT = &(scratch_wT[2*C*K]); 106 element_filter_type *scratch_woT = &(scratch_wT[3*C*K]);
|
H A D | libxsmm_dnn_rnncell_st_rnn_bwdupd_nc_ck_generic.tpl.c | 35 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; variable 50 LIBXSMM_VLA_DECL(2, element_filter_type, wT, scratch_wT, C);
|
H A D | libxsmm_dnn_rnncell_st_rnn_bwdupd_nc_kcck.tpl.c | 36 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; 59 LIBXSMM_VLA_DECL(4, element_filter_type, wT, scratch_wT, kBlocks, bk, bc);
|
H A D | libxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c | 48 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; variable 67 element_filter_type *scratch_wiT = &(scratch_wT[0]); 68 element_filter_type *scratch_wcT = &(scratch_wT[C*K]); 69 element_filter_type *scratch_wfT = &(scratch_wT[2*C*K]);
|
H A D | libxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c | 49 element_filter_type *scratch_wT = (element_filter_type*)handle->scratch_wT; variable 76 element_filter_type *scratch_wiT = &(scratch_wT[0]); 77 element_filter_type *scratch_wcT = &(scratch_wT[C*K]); 78 element_filter_type *scratch_wfT = &(scratch_wT[2*C*K]);
|
/dports/math/libxsmm/libxsmm-1.16.3/src/ |
H A D | libxsmm_dnn_rnncell.c | 144 handle->scratch_wT = 0; in libxsmm_dnn_create_rnncell() 1042 handle->scratch_wT = (void*)address; in libxsmm_dnn_rnncell_bind_scratch() 1045 handle->scratch_wT = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch() 1205 handle->scratch_wT = (void*)address; in libxsmm_dnn_rnncell_bind_scratch() 1208 handle->scratch_wT = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch() 1472 handle->scratch_wT = (void*)address; in libxsmm_dnn_rnncell_bind_scratch() 1475 handle->scratch_wT = (void*)(address+offset); in libxsmm_dnn_rnncell_bind_scratch() 1625 handle->scratch_wT = 0; in libxsmm_dnn_rnncell_release_scratch() 1656 handle->scratch_wT = 0; in libxsmm_dnn_rnncell_release_scratch() 1702 handle->scratch_wT = 0; in libxsmm_dnn_rnncell_release_scratch()
|
H A D | libxsmm_main.h | 698 void* scratch_wT; member
|