Searched refs:gemm_ukernel (Results 1 – 2 of 2) sorted by relevance
244 struct xnn_hmp_gemm_ukernel gemm_ukernel = fully_connected_op->ukernel.gemm.gemm_cases[mr-1]; in setup_fully_connected_nc() local246 gemm_ukernel = fully_connected_op->ukernel.gemm.gemm_cases[0]; in setup_fully_connected_nc()261 .ukernel = gemm_ukernel, in setup_fully_connected_nc()280 if (xnn_is_hmp_gemm_ukernel(gemm_ukernel)) { in setup_fully_connected_nc()
1453 struct xnn_hmp_gemm_ukernel gemm_ukernel = gemm_cases[mr - 1]; in setup_convolution2d_nhwc() local1467 .ukernel = gemm_ukernel, in setup_convolution2d_nhwc()1491 if (xnn_is_hmp_gemm_ukernel(gemm_ukernel)) { in setup_convolution2d_nhwc()1508 if (xnn_is_hmp_gemm_ukernel(gemm_ukernel)) { in setup_convolution2d_nhwc()