/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/tf_lstm_ops/ |
H A D | lstm_bwd.c | 234 …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 D | libxsmm_dnn_tensor.h | 142 LIBXSMM_DNN_RNN_GRADIENT_WEIGHT, enumerator
|
/dports/math/libxsmm/libxsmm-1.16.3/src/ |
H A D | libxsmm_dnn_rnncell.c | 311 …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 D | grudriver_nc_ck.c | 343 …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 D | grudriver_nc_kcck.c | 347 …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 D | rnndriver_nc_ck_f32.c | 384 …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 D | rnndriver_nc_kcck_f32.c | 389 …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 D | rnndriver_ncnc_kcck_f32.c | 389 …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 D | lstmdriver_nc_ck_f32.c | 400 …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 D | lstmdriver_nc_kcck_bf16.c | 471 …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 D | lstmdriver_nc_ck_bf16.c | 457 …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 D | lstmdriver_nc_kcck_f32.c | 646 …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()
|