Searched refs:gemm_ukernel (Results 1 – 2 of 2) sorted by relevance
199 struct xnn_hmp_gemm_ukernel gemm_ukernel = fully_connected_op->ukernel.gemm.general_case; in setup_fully_connected_nc() local201 gemm_ukernel = fully_connected_op->ukernel.gemm.mr1_case; in setup_fully_connected_nc()215 .ukernel = gemm_ukernel, in setup_fully_connected_nc()
859 struct xnn_hmp_gemm_ukernel gemm_ukernel = convolution_op->ukernel.gemm.general_case; in setup_convolution2d_nhwc() local862 gemm_ukernel = convolution_op->ukernel.gemm.mr1_case; in setup_convolution2d_nhwc()877 .ukernel = gemm_ukernel, in setup_convolution2d_nhwc()892 if (xnn_is_hmp_gemm_ukernel(gemm_ukernel)) { in setup_convolution2d_nhwc()909 if (xnn_is_hmp_gemm_ukernel(gemm_ukernel)) { in setup_convolution2d_nhwc()