Searched refs:libxsmm_dnn_get_feature_map_blocks (Results 1 – 9 of 9) sorted by relevance
| /dports/math/libxsmm/libxsmm-1.16.3/src/ |
| H A D | libxsmm_dnn_optimizer.c | 33 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.K, in libxsmm_dnn_create_optimizer()
|
| H A D | libxsmm_dnn_softmaxloss.c | 37 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.C, in libxsmm_dnn_create_softmaxloss()
|
| H A D | libxsmm_dnn_pooling.c | 34 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.C, in libxsmm_dnn_create_pooling()
|
| H A D | libxsmm_dnn_fusedbatchnorm.c | 43 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.C, in libxsmm_dnn_create_fusedbatchnorm()
|
| H A D | libxsmm_dnn.c | 37 LIBXSMM_API_INTERN libxsmm_dnn_err_t libxsmm_dnn_get_feature_map_blocks( int C, int K, int* C_block… in libxsmm_dnn_get_feature_map_blocks() function
|
| H A D | libxsmm_main.h | 958 LIBXSMM_API_INTERN libxsmm_dnn_err_t libxsmm_dnn_get_feature_map_blocks(
|
| H A D | libxsmm_dnn_fusedgroupnorm.c | 34 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.C, in libxsmm_dnn_create_fusedgroupnorm()
|
| H A D | libxsmm_dnn_fullyconnected.c | 528 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.K, in libxsmm_dnn_create_fullyconnected()
532 *status = libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.K, in libxsmm_dnn_create_fullyconnected()
|
| H A D | libxsmm_dnn_convolution.c | 41 …libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.K, &result, &ofm, &lp, handle->de… in libxsmm_dnn_convolution_setup_ifmblock()
50 …libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.K, &ifm, &result, &lp, handle->de… in libxsmm_dnn_convolution_setup_ofmblock()
59 …libxsmm_dnn_get_feature_map_blocks( handle->desc.C, handle->desc.K, &ifm, &ofm, &result, handle->d… in libxsmm_dnn_convolution_setup_fm_lp_block()
|