Searched refs:mr1_function (Results 1 – 4 of 4) sorted by relevance
117 xnn_gemm_ukernel_function mr1_function; member125 xnn_igemm_ukernel_function mr1_function; member
314 .mr1_function = xnn_params.f32.gemm.gemm1, in xnn_create_fully_connected_nc_f32()370 if (batch_size == 1 && fully_connected_op->ukernel.gemm.mr1_function != NULL) { in setup_fully_connected_nc()371 gemm_ukernel = fully_connected_op->ukernel.gemm.mr1_function; in setup_fully_connected_nc()
640 .mr1_function = xnn_params.f32.gemm.gemm1, in xnn_create_convolution2d_nhwc_f32()660 .mr1_function = xnn_params.f32.gemm.igemm1, in xnn_create_convolution2d_nhwc_f32()805 if (batch_output_size == 1 && convolution_op->ukernel.gemm.mr1_function != NULL) { in setup_convolution2d_nhwc()807 gemm_ukernel = convolution_op->ukernel.gemm.mr1_function; in setup_convolution2d_nhwc()868 if (output_size == 1 && convolution_op->ukernel.igemm.mr1_function != NULL) { in setup_convolution2d_nhwc()870 igemm_ukernel = convolution_op->ukernel.igemm.mr1_function; in setup_convolution2d_nhwc()
615 if (output_size == 1 && deconvolution_op->ukernel.igemm.mr1_function != NULL) { in setup_conv_path()616 deconvolution_op->context.igemm.ukernel = deconvolution_op->ukernel.igemm.mr1_function; in setup_conv_path()