Searched refs:mr1_case (Results 1 – 4 of 4) sorted by relevance
142 .mr1_case = gemm_ukernels->gemm1, in create_fully_connected_nc()200 …if (batch_size == 1 && fully_connected_op->ukernel.gemm.mr1_case.function[XNN_UARCH_DEFAULT] != NU… in setup_fully_connected_nc()201 gemm_ukernel = fully_connected_op->ukernel.gemm.mr1_case; in setup_fully_connected_nc()
338 .mr1_case = gemm_ukernels->gemm1, in create_convolution2d_nhwc()358 .mr1_case = gemm_ukernels->igemm1, in create_convolution2d_nhwc()860 …if (batch_output_size == 1 && convolution_op->ukernel.gemm.mr1_case.function[XNN_UARCH_DEFAULT] !=… in setup_convolution2d_nhwc()862 gemm_ukernel = convolution_op->ukernel.gemm.mr1_case; in setup_convolution2d_nhwc()943 …if (output_size == 1 && convolution_op->ukernel.igemm.mr1_case.function[XNN_UARCH_DEFAULT] != NULL… in setup_convolution2d_nhwc()945 igemm_ukernel = convolution_op->ukernel.igemm.mr1_case; in setup_convolution2d_nhwc()
544 …if (output_size == 1 && deconvolution_op->ukernel.igemm.mr1_case.function[XNN_UARCH_DEFAULT] != NU… in setup_conv_path()545 deconvolution_op->context.igemm.ukernel = deconvolution_op->ukernel.igemm.mr1_case; in setup_conv_path()
123 struct xnn_hmp_gemm_ukernel mr1_case; member131 struct xnn_hmp_igemm_ukernel mr1_case; member