Home
last modified time | relevance | path

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

/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_rnncell_forward.c45 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
79 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
116 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
155 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
189 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
223 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
260 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
331 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_st_fwd_nc_ck()
388 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_st_fwd_ncnc_kcck()
456 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_st_fwd_nc_kcck()
H A Dlibxsmm_dnn_rnncell_backward_weight_update.c46 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
80 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
117 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
154 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
192 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
232 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in LIBXSMM_INTRINSICS()
334 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_st_bwdupd_nc_ck()
414 } else if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_st_bwdupd_nc_kcck()
H A Dlibxsmm_dnn_rnncell.c519 if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
532 if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
771 if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
784 if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
832 if ( handle->desc.cell_type == LIBXSMM_DNN_RNNCELL_LSTM ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
909 case LIBXSMM_DNN_RNNCELL_LSTM: { in libxsmm_dnn_rnncell_get_scratch_size()
1086 case LIBXSMM_DNN_RNNCELL_LSTM: { in libxsmm_dnn_rnncell_bind_scratch()
1636 case LIBXSMM_DNN_RNNCELL_LSTM: { in libxsmm_dnn_rnncell_release_scratch()
1767 case LIBXSMM_DNN_RNNCELL_LSTM: { in libxsmm_dnn_rnncell_get_internalstate_size()
1865 case LIBXSMM_DNN_RNNCELL_LSTM: { in libxsmm_dnn_rnncell_bind_internalstate()
[all …]
/dports/math/libxsmm/libxsmm-1.16.3/include/
H A Dlibxsmm_dnn_rnncell.h28 LIBXSMM_DNN_RNNCELL_LSTM, enumerator
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tf_lstm_ops/
H A Dlstm_fwd.c131 lstmcell_desc.cell_type = LIBXSMM_DNN_RNNCELL_LSTM; in lstm_fwd_create()
H A Dlstm_bwd.c147 lstmcell_desc.cell_type = LIBXSMM_DNN_RNNCELL_LSTM; in lstm_bwd_create()
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/lstmdriver/
H A Dlstmdriver_nc_ck_f32.c325 lstmcell_desc.cell_type = LIBXSMM_DNN_RNNCELL_LSTM; in main()
H A Dlstmdriver_nc_kcck_bf16.c388 lstmcell_desc.cell_type = LIBXSMM_DNN_RNNCELL_LSTM; in main()
H A Dlstmdriver_nc_ck_bf16.c382 lstmcell_desc.cell_type = LIBXSMM_DNN_RNNCELL_LSTM; in main()
H A Dlstmdriver_nc_kcck_f32.c564 lstmcell_desc.cell_type = LIBXSMM_DNN_RNNCELL_LSTM; in main()