Searched refs:rnncell_desc (Results 1 – 5 of 5) sorted by relevance
76 libxsmm_dnn_rnncell_desc rnncell_desc; in main() local329 rnncell_desc.threads = nThreads; in main()330 rnncell_desc.N = N; in main()331 rnncell_desc.C = C; in main()332 rnncell_desc.K = K; in main()333 rnncell_desc.bn = bn; in main()334 rnncell_desc.bk = bk; in main()335 rnncell_desc.bc = bc; in main()336 rnncell_desc.max_T = t; in main()347 rnncell_desc.datatype_in = LIBXSMM_DNN_DATATYPE_F32; in main()[all …]
76 libxsmm_dnn_rnncell_desc rnncell_desc; in main() local326 rnncell_desc.threads = nThreads; in main()327 rnncell_desc.N = N; in main()328 rnncell_desc.C = C; in main()329 rnncell_desc.K = K; in main()330 rnncell_desc.bn = bn; in main()331 rnncell_desc.bk = bk; in main()332 rnncell_desc.bc = bc; in main()333 rnncell_desc.max_T = t; in main()344 rnncell_desc.datatype_in = LIBXSMM_DNN_DATATYPE_F32; in main()[all …]
49 LIBXSMM_API libxsmm_dnn_rnncell* libxsmm_dnn_create_rnncell(libxsmm_dnn_rnncell_desc rnncell_desc, …
24 LIBXSMM_API libxsmm_dnn_rnncell* libxsmm_dnn_create_rnncell(libxsmm_dnn_rnncell_desc rnncell_desc, … in libxsmm_dnn_create_rnncell() argument32 if ( (rnncell_desc.datatype_in != rnncell_desc.datatype_out) || in libxsmm_dnn_create_rnncell()33 …( (rnncell_desc.datatype_in != LIBXSMM_DNN_DATATYPE_BF16) && (rnncell_desc.datatype_in != LIBXSMM_… in libxsmm_dnn_create_rnncell()38 …if ( (rnncell_desc.datatype_in == LIBXSMM_DNN_DATATYPE_BF16) || (rnncell_desc.datatype_out == LIBX… in libxsmm_dnn_create_rnncell()39 … (LIBXSMM_X86_AVX512_CORE > libxsmm_target_archid) || (rnncell_desc.C % 16 != 0) || (rnncell_desc.… in libxsmm_dnn_create_rnncell()45 if (rnncell_desc.max_T < 1) { in libxsmm_dnn_create_rnncell()56 handle->desc = rnncell_desc; in libxsmm_dnn_create_rnncell()58 handle->T = rnncell_desc.max_T; in libxsmm_dnn_create_rnncell()