Lines Matching refs:gemm
132 const uint32_t nr = xnn_params.q8.gemm.nr; in xnn_create_fully_connected_nc_q8()
133 const uint32_t kr = UINT32_C(1) << xnn_params.q8.gemm.log2_kr; in xnn_create_fully_connected_nc_q8()
177 fully_connected_op->ukernel.gemm = (struct xnn_ukernel_gemm) { in xnn_create_fully_connected_nc_q8()
178 .default_function = xnn_params.q8.gemm.gemm, in xnn_create_fully_connected_nc_q8()
179 .mr = xnn_params.q8.gemm.mr, in xnn_create_fully_connected_nc_q8()
273 const uint32_t nr = xnn_params.f32.gemm.nr; in xnn_create_fully_connected_nc_f32()
274 const uint32_t kr = UINT32_C(1) << xnn_params.f32.gemm.log2_kr; in xnn_create_fully_connected_nc_f32()
275 const uint32_t sr = UINT32_C(1) << xnn_params.f32.gemm.log2_sr; in xnn_create_fully_connected_nc_f32()
312 fully_connected_op->ukernel.gemm = (struct xnn_ukernel_gemm) { in xnn_create_fully_connected_nc_f32()
313 .default_function = xnn_params.f32.gemm.gemm, in xnn_create_fully_connected_nc_f32()
314 .mr1_function = xnn_params.f32.gemm.gemm1, in xnn_create_fully_connected_nc_f32()
315 .mr = xnn_params.f32.gemm.mr, in xnn_create_fully_connected_nc_f32()
366 uint32_t mr = fully_connected_op->ukernel.gemm.mr; in setup_fully_connected_nc()
367 const uint32_t nr = fully_connected_op->ukernel.gemm.nr; in setup_fully_connected_nc()
369 xnn_gemm_ukernel_function gemm_ukernel = fully_connected_op->ukernel.gemm.default_function; in setup_fully_connected_nc()
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()
375 fully_connected_op->context.gemm = (struct gemm_context) { in setup_fully_connected_nc()
377 ….w_stride = (round_up_po2(input_channels, fully_connected_op->ukernel.gemm.kr) << log2_input_eleme… in setup_fully_connected_nc()
387 …memcpy(&fully_connected_op->context.gemm.params, params, sizeof(fully_connected_op->context.gemm.p… in setup_fully_connected_nc()