Home
last modified time | relevance | path

Searched refs:upd_2d_blocking (Results 1 – 5 of 5) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_fullyconnected.c75 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
95 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
112 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
129 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
146 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
163 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
180 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
197 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
213 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
229 handle->upd_2d_blocking = 0; in libxsmm_dnn_create_fullyconnected()
[all …]
H A Dlibxsmm_main.h612 int upd_2d_blocking; member
/dports/math/libxsmm/libxsmm-1.16.3/src/template/
H A Dlibxsmm_dnn_fullyconnected_st_bwdupd_ncnc_kcck_generic.tpl.c241 const int ofm_subtasks = (handle->upd_2d_blocking == 1) ? 1 : handle->ofm_subtasks;
242 const int ifm_subtasks = (handle->upd_2d_blocking == 1) ? 1 : handle->ifm_subtasks;
243 const int bbk = (handle->upd_2d_blocking == 1) ? bk : bk/ofm_subtasks;
244 const int bbc = (handle->upd_2d_blocking == 1) ? bc : bc/ifm_subtasks;
250 int use_2d_blocking = handle->upd_2d_blocking;
H A Dlibxsmm_dnn_fullyconnected_st_bwdupd_ncnc_kcck_generic_bf16.tpl.c381 const int ofm_subtasks = (handle->upd_2d_blocking == 1) ? 1 : handle->ofm_subtasks;
382 const int ifm_subtasks = (handle->upd_2d_blocking == 1) ? 1 : handle->ifm_subtasks;
383 const int bbk = (handle->upd_2d_blocking == 1) ? bk : bk/ofm_subtasks;
384 const int bbc = (handle->upd_2d_blocking == 1) ? bc : bc/ifm_subtasks;
390 int use_2d_blocking = handle->upd_2d_blocking;
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/fullyconnecteddriver/
H A Dlayer_example_bf16.c115 int upd_2d_blocking = atoi(getenv("UPD_2D_BLOCKING")); in main()
130 int upd_2d_blocking = 1; in main() local
639 …_BN=%d_BK=%d_BC=%d_BFACCUM=%d_IFMSUBTASK=%d_OFMSUBTASK=%d",nThreads, upd_2d_blocking, upd_row_team… in main()