Home
last modified time | relevance | path

Searched refs:LIBXSMM_DNN_RNN_GRADIENT_WEIGHT (Results 1 – 12 of 12) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tf_lstm_ops/
H A Dlstm_bwd.c234 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in lstm_bwd_create()
282 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in lstm_bwd_create()
373 …_set_tensor_data_ptr( libxsmm_dnn_rnncell_get_tensor(handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in lstm_bwd_set_ptr()
440 …m_dnn_destroy_tensor( libxsmm_dnn_rnncell_get_tensor(handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in lstm_bwd_destroy()
457 …CHKERR_LIBXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) … in lstm_bwd_destroy()
/dports/math/libxsmm/libxsmm-1.16.3/include/
H A Dlibxsmm_dnn_tensor.h142 LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, enumerator
/dports/math/libxsmm/libxsmm-1.16.3/src/
H A Dlibxsmm_dnn_rnncell.c311 …if ( (type == LIBXSMM_DNN_RNN_REGULAR_WEIGHT) || (type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) || in libxsmm_dnn_rnncell_create_tensor_datalayout()
325 … if ( (type == LIBXSMM_DNN_RNN_REGULAR_WEIGHT) || (type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
374 … if ( (type == LIBXSMM_DNN_RNN_REGULAR_WEIGHT) || (type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
414 … if ( (type == LIBXSMM_DNN_RNN_REGULAR_WEIGHT) || (type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
467 … if ( (type == LIBXSMM_DNN_RNN_REGULAR_WEIGHT) || (type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
516 … if ( (type == LIBXSMM_DNN_RNN_REGULAR_WEIGHT) || (type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) ) { in libxsmm_dnn_rnncell_create_tensor_datalayout()
1991 …pe != LIBXSMM_DNN_RNN_REGULAR_WEIGHT) && (type != LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) … in libxsmm_dnn_rnncell_bind_tensor()
2024 } else if ( type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) { in libxsmm_dnn_rnncell_bind_tensor()
2080 …pe != LIBXSMM_DNN_RNN_REGULAR_WEIGHT) && (type != LIBXSMM_DNN_RNN_GRADIENT_WEIGHT) … in libxsmm_dnn_rnncell_get_tensor()
2109 } else if ( type == LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) { in libxsmm_dnn_rnncell_get_tensor()
[all …]
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/grudriver/
H A Dgrudriver_nc_ck.c343 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
386 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
846 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
H A Dgrudriver_nc_kcck.c347 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
392 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
864 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/rnndriver/
H A Drnndriver_nc_ck_f32.c384 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
416 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
884 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
H A Drnndriver_nc_kcck_f32.c389 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
426 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
888 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
H A Drnndriver_ncnc_kcck_f32.c389 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
426 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
848 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/lstmdriver/
H A Dlstmdriver_nc_ck_f32.c400 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
456 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
974 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
H A Dlstmdriver_nc_kcck_bf16.c471 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
539 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
1090 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
H A Dlstmdriver_nc_ck_bf16.c457 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
513 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
1048 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
H A Dlstmdriver_nc_kcck_f32.c646 …ibxsmm_dnn_rnncell_create_tensor_datalayout( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, &sta… in main()
714 …xsmm_dnn_rnncell_bind_tensor( libxsmm_handle, libxsmm_dweight, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()
1246 …BXSMM_DNN( libxsmm_dnn_rnncell_release_tensor( libxsmm_handle, LIBXSMM_DNN_RNN_GRADIENT_WEIGHT ) ); in main()