Home
last modified time | relevance | path

Searched refs:gemm_ukernels (Results 1 – 3 of 3) sorted by relevance

/external/XNNPACK/src/operators/
Dfully-connected-nc.c43 const struct gemm_fused_ukernels* gemm_ukernels, in create_fully_connected_nc() argument
141 .general_case = gemm_ukernels->gemm, in create_fully_connected_nc()
142 .mr1_case = gemm_ukernels->gemm1, in create_fully_connected_nc()
349 const struct gemm_fused_ukernels* gemm_ukernels = &xnn_params.f32.gemm.minmax; in xnn_create_fully_connected_nc_f32() local
352 gemm_ukernels = &xnn_params.f32.gemm.linear; in xnn_create_fully_connected_nc_f32()
366 &xnn_params.f32.gemm, gemm_ukernels, in xnn_create_fully_connected_nc_f32()
Ddeconvolution-nhwc.c71 const struct gemm_fused_ukernels* gemm_ukernels, in create_deconvolution2d_nhwc()
266 .general_case = gemm_ukernels->igemm, in create_deconvolution2d_nhwc()
267 .gemm_case = gemm_ukernels->gemm, in create_deconvolution2d_nhwc()
445 const struct gemm_fused_ukernels* gemm_ukernels = &gemm_parameters->minmax; in xnn_create_deconvolution2d_nhwc_f32() local
448 gemm_ukernels = &gemm_parameters->linear; in xnn_create_deconvolution2d_nhwc_f32()
467 gemm_parameters, gemm_ukernels, in xnn_create_deconvolution2d_nhwc_f32()
Dconvolution-nhwc.c321 const struct gemm_fused_ukernels* gemm_ukernels = &gemm_parameters->minmax; in create_convolution2d_nhwc() local
323 gemm_ukernels = &gemm_parameters->linear; in create_convolution2d_nhwc()
325 gemm_ukernels = &gemm_parameters->relu; in create_convolution2d_nhwc()
337 .general_case = gemm_ukernels->gemm, in create_convolution2d_nhwc()
338 .mr1_case = gemm_ukernels->gemm1, in create_convolution2d_nhwc()
357 .general_case = gemm_ukernels->igemm, in create_convolution2d_nhwc()
358 .mr1_case = gemm_ukernels->igemm1, in create_convolution2d_nhwc()