Searched refs:libxsmm_dnn_rnncell_get_scratch_size (Results 1 – 13 of 13) sorted by relevance
/dports/math/libxsmm/libxsmm-1.16.3/include/ |
H A D | libxsmm_dnn_rnncell.h | 54 LIBXSMM_API size_t libxsmm_dnn_rnncell_get_scratch_size(const libxsmm_dnn_rnncell* handle, const li…
|
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tf_lstm_ops/ |
H A D | lstm_fwd.c | 223 …size_t scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KI… in lstm_fwd_create() 343 …size_t scratch_size = libxsmm_dnn_rnncell_get_scratch_size( handle, LIBXSMM_DNN_COMPUTE_KIND_FWD, … in lstm_fwd_destroy()
|
H A D | lstm_bwd.c | 288 …size_t scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KI… in lstm_bwd_create() 463 …size_t scratch_size = libxsmm_dnn_rnncell_get_scratch_size( handle, LIBXSMM_DNN_COMPUTE_KIND_BWDUP… in lstm_bwd_destroy()
|
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/grudriver/ |
H A D | grudriver_nc_ck.c | 393 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 398 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
H A D | grudriver_nc_kcck.c | 399 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 404 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/rnndriver/ |
H A D | rnndriver_nc_ck_f32.c | 423 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 428 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
H A D | rnndriver_nc_kcck_f32.c | 433 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 438 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
H A D | rnndriver_ncnc_kcck_f32.c | 433 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 438 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/lstmdriver/ |
H A D | lstmdriver_nc_ck_f32.c | 464 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 469 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
H A D | lstmdriver_nc_kcck_bf16.c | 547 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 552 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
H A D | lstmdriver_nc_ck_bf16.c | 521 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 526 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
H A D | lstmdriver_nc_kcck_f32.c | 722 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_FWD,… in main() 727 …scratch_size = libxsmm_dnn_rnncell_get_scratch_size( libxsmm_handle, LIBXSMM_DNN_COMPUTE_KIND_ALL,… in main()
|
/dports/math/libxsmm/libxsmm-1.16.3/src/ |
H A D | libxsmm_dnn_rnncell.c | 876 LIBXSMM_API size_t libxsmm_dnn_rnncell_get_scratch_size(const libxsmm_dnn_rnncell* handle, const li… in libxsmm_dnn_rnncell_get_scratch_size() function
|