Home
last modified time | relevance | path

Searched refs:cogoldt (Results 1 – 4 of 4) sorted by relevance

/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/lstmdriver/
H A Dlstmdriver_nc_kcck_bf16.c48 float *xgoldt, *cspgold,*hpgold, *csgoldt, *cogoldt, *hgoldt; in main() local
185 cogoldt = (float*)libxsmm_aligned_malloc(K*N*t*sizeof(float), 2097152); in main()
256 zero_buf(cogoldt, K*N*t); in main()
309 rne_mask_fp32_bf16( cogoldt, cogoldt, N*K*t ); in main()
331 csgoldt, cogoldt, hgoldt, in main()
337 rne_mask_fp32_bf16( cogoldt, cogoldt, K*N*t ); in main()
346 csgoldt, cogoldt, hgoldt, in main()
843 csgoldt, cogoldt, hgoldt, in main()
1034 csgoldt, cogoldt, hgoldt, in main()
1138 libxsmm_free(cogoldt); in main()
H A Dlstmdriver_nc_ck_bf16.c48 float *xgoldt, *cspgold,*hpgold, *csgoldt, *cogoldt, *hgoldt; in main() local
183 cogoldt = (float*)libxsmm_aligned_malloc(K*N*t*sizeof(float), 2097152); in main()
250 zero_buf(cogoldt, K*N*t); in main()
303 rne_mask_fp32_bf16( cogoldt, cogoldt, N*K*t ); in main()
325 csgoldt, cogoldt, hgoldt, in main()
331 rne_mask_fp32_bf16( cogoldt, cogoldt, K*N*t ); in main()
340 csgoldt, cogoldt, hgoldt, in main()
804 csgoldt, cogoldt, hgoldt, in main()
995 csgoldt, cogoldt, hgoldt, in main()
1094 libxsmm_free(cogoldt); in main()
H A Dlstmdriver_nc_ck_f32.c43 float *xgoldt, *cspgold,*hpgold, *csgoldt, *cogoldt, *hgoldt; in main() local
177 cogoldt = (float*)libxsmm_aligned_malloc(K*N*t*sizeof(float), 2097152); in main()
238 zero_buf(cogoldt, K*N*t); in main()
284 csgoldt, cogoldt, hgoldt, in main()
289 csgoldt, cogoldt, hgoldt, in main()
721 csgoldt, cogoldt, hgoldt, in main()
919 csgoldt, cogoldt, hgoldt, in main()
1020 libxsmm_free(cogoldt); in main()
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/common/
H A Ddnn_common.h2735 float *csgoldt, float *cogoldt, float *hgoldt, in lstm_ref_fwd() argument
2748 LIBXSMM_VLA_DECL(2, float, cogold, cogoldt, K * N); in lstm_ref_fwd()
2844 float *csgoldt, float *cogoldt, float *hgoldt, in lstm_ref_bwd_upd() argument
2865 LIBXSMM_VLA_DECL(2, float, cogold, cogoldt, K * N); in lstm_ref_bwd_upd()