Searched refs:gemm_parameters (Results 1 – 4 of 4) sorted by relevance
42 const struct gemm_parameters* gemm_parameters, in create_fully_connected_nc() argument98 const uint32_t nr = gemm_parameters->nr; in create_fully_connected_nc()99 const uint32_t kr = UINT32_C(1) << gemm_parameters->log2_kr; in create_fully_connected_nc()100 const uint32_t sr = UINT32_C(1) << gemm_parameters->log2_sr; in create_fully_connected_nc()143 .mr = gemm_parameters->mr, in create_fully_connected_nc()
70 const struct gemm_parameters* gemm_parameters, in create_deconvolution2d_nhwc()168 const uint32_t mr = gemm_parameters->mr; in create_deconvolution2d_nhwc()169 const uint32_t nr = gemm_parameters->nr; in create_deconvolution2d_nhwc()170 const uint32_t kr = UINT32_C(1) << gemm_parameters->log2_kr; in create_deconvolution2d_nhwc()171 const uint32_t sr = UINT32_C(1) << gemm_parameters->log2_sr; in create_deconvolution2d_nhwc()438 const struct gemm_parameters* gemm_parameters = &xnn_params.f32.gemm; in xnn_create_deconvolution2d_nhwc_f32() local439 if (gemm_parameters->nr > group_output_channels) { in xnn_create_deconvolution2d_nhwc_f32()442 gemm_parameters = &xnn_params.f32.gemm2; in xnn_create_deconvolution2d_nhwc_f32()445 const struct gemm_fused_ukernels* gemm_ukernels = &gemm_parameters->minmax; in xnn_create_deconvolution2d_nhwc_f32()447 if (linear_activation && gemm_parameters->linear.gemm.function[XNN_UARCH_DEFAULT] != NULL) { in xnn_create_deconvolution2d_nhwc_f32()[all …]
94 const struct gemm_parameters* gemm_parameters, in create_convolution2d_nhwc()305 const uint32_t nr = gemm_parameters->nr; in create_convolution2d_nhwc()306 const uint32_t kr = UINT32_C(1) << gemm_parameters->log2_kr; in create_convolution2d_nhwc()307 const uint32_t sr = UINT32_C(1) << gemm_parameters->log2_sr; in create_convolution2d_nhwc()321 const struct gemm_fused_ukernels* gemm_ukernels = &gemm_parameters->minmax; in create_convolution2d_nhwc()322 if (linear_activation && gemm_parameters->linear.gemm.function[XNN_UARCH_DEFAULT] != NULL) { in create_convolution2d_nhwc()323 gemm_ukernels = &gemm_parameters->linear; in create_convolution2d_nhwc()324 … } else if (relu_activation && gemm_parameters->relu.gemm.function[XNN_UARCH_DEFAULT] != NULL) { in create_convolution2d_nhwc()325 gemm_ukernels = &gemm_parameters->relu; in create_convolution2d_nhwc()334 .mr = gemm_parameters->mr, in create_convolution2d_nhwc()[all …]
1797 struct gemm_parameters { struct2008 struct gemm_parameters gemm;2014 struct gemm_parameters gemm;2032 struct gemm_parameters gemm;2033 struct gemm_parameters gemm2;2041 struct gemm_parameters gemm;2042 struct gemm_parameters gemm2;