Home
last modified time | relevance | path

Searched refs:scratch_wT (Results 1 – 10 of 10) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck.tpl.c57 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 Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic.tpl.c57 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 Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_ck_generic_bf16.tpl.c64 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 Dlibxsmm_dnn_rnncell_st_lstm_bwdupd_nc_kcck_bf16.tpl.c64 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 Dlibxsmm_dnn_rnncell_st_rnn_bwdupd_nc_ck_generic.tpl.c35 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 Dlibxsmm_dnn_rnncell_st_rnn_bwdupd_nc_kcck.tpl.c36 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 Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_kcck.tpl.c48 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 Dlibxsmm_dnn_rnncell_st_gru_bwdupd_nc_ck_generic.tpl.c49 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 Dlibxsmm_dnn_rnncell.c144 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 Dlibxsmm_main.h698 void* scratch_wT; member