Searched refs:cogoldt (Results 1 – 4 of 4) sorted by relevance
/dports/math/libxsmm/libxsmm-1.16.3/samples/deeplearning/lstmdriver/ |
H A D | lstmdriver_nc_kcck_bf16.c | 48 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 D | lstmdriver_nc_ck_bf16.c | 48 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 D | lstmdriver_nc_ck_f32.c | 43 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 D | dnn_common.h | 2735 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()
|