Searched defs:gemm_cases (Results 1 – 3 of 3) sorted by relevance
81 size_t batch_size, uint32_t max_mr, uint32_t nr, struct xnn_hmp_gemm_ukernel *gemm_cases) in xnn_get_heuristic_mr_gemm()
51 struct xnn_hmp_gemm_ukernel gemm_cases[XNN_MAX_MR]; member60 struct xnn_hmp_gemm_ukernel gemm_cases[XNN_MAX_MR]; member
1434 struct xnn_hmp_gemm_ukernel *gemm_cases = convolution_op->ukernel.gemm.gemm_cases; in setup_convolution2d_nhwc() local