Home
last modified time | relevance | path

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 Dlibxsmm_dnn_rnncell.h54 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 Dlstm_fwd.c223 …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 Dlstm_bwd.c288 …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 Dgrudriver_nc_ck.c393 …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 Dgrudriver_nc_kcck.c399 …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 Drnndriver_nc_ck_f32.c423 …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 Drnndriver_nc_kcck_f32.c433 …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 Drnndriver_ncnc_kcck_f32.c433 …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 Dlstmdriver_nc_ck_f32.c464 …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 Dlstmdriver_nc_kcck_bf16.c547 …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 Dlstmdriver_nc_ck_bf16.c521 …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 Dlstmdriver_nc_kcck_f32.c722 …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 Dlibxsmm_dnn_rnncell.c876 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