Home
last modified time | relevance | path

Searched refs:igemm_cases (Results 1 – 4 of 4) sorted by relevance

/external/XNNPACK/src/
Doperator-utils.c109 size_t batch_size, uint32_t max_mr, uint32_t nr, struct xnn_hmp_igemm_ukernel *igemm_cases) in xnn_get_heuristic_mr_igemm() argument
111 assert(igemm_cases[max_mr-1].function[XNN_UARCH_DEFAULT] != NULL); in xnn_get_heuristic_mr_igemm()
112 if (batch_size <= max_mr && igemm_cases[batch_size-1].function[XNN_UARCH_DEFAULT] != NULL) { in xnn_get_heuristic_mr_igemm()
123 if (igemm_cases[mr-1].function[XNN_UARCH_DEFAULT] == NULL) { in xnn_get_heuristic_mr_igemm()
132 assert(igemm_cases[best_mr-1].function[XNN_UARCH_DEFAULT] != NULL); in xnn_get_heuristic_mr_igemm()
/external/XNNPACK/src/xnnpack/
Doperator.h59 struct xnn_hmp_igemm_ukernel igemm_cases[XNN_MAX_MR]; member
358 struct xnn_hmp_igemm_ukernel *igemm_cases);
/external/XNNPACK/src/operators/
Dconvolution-nhwc.c186 convolution_op->ukernel.igemm.igemm_cases[0].generated_code_offset[XNN_UARCH_DEFAULT] = in generate_igemms_up_to_max_mr()
190 convolution_op->ukernel.igemm.igemm_cases[mr - 1].generated_code_offset[XNN_UARCH_DEFAULT] = in generate_igemms_up_to_max_mr()
565 convolution_op->ukernel.igemm.igemm_cases[i] = gemm_ukernels->igemm[i]; in create_convolution2d_nhwc()
1541 struct xnn_hmp_igemm_ukernel* igemm_cases = convolution_op->ukernel.igemm.igemm_cases; in setup_convolution2d_nhwc() local
1544 mr = xnn_get_heuristic_mr_igemm(output_size, mr, nr, igemm_cases); in setup_convolution2d_nhwc()
1546 if (output_size == 1 && igemm_cases[0].function[XNN_UARCH_DEFAULT] != NULL) { in setup_convolution2d_nhwc()
1553 … const size_t jit_code_offset = igemm_cases[mr - 1].generated_code_offset[XNN_UARCH_DEFAULT]; in setup_convolution2d_nhwc()
1555 igemm_cases[mr - 1].function[XNN_UARCH_DEFAULT] = in setup_convolution2d_nhwc()
1560 struct xnn_hmp_igemm_ukernel igemm_ukernel = igemm_cases[mr - 1]; in setup_convolution2d_nhwc()
Ddeconvolution-nhwc.c274 deconvolution_op->ukernel.igemm.igemm_cases[i] = gemm_ukernels->igemm[i]; in create_deconvolution2d_nhwc()
689 struct xnn_hmp_igemm_ukernel igemm_ukernel = deconvolution_op->ukernel.igemm.igemm_cases[mr - 1]; in setup_conv_path()
690 …if (output_size == 1 && deconvolution_op->ukernel.igemm.igemm_cases[0].function[XNN_UARCH_DEFAULT]… in setup_conv_path()
692 igemm_ukernel = deconvolution_op->ukernel.igemm.igemm_cases[0]; in setup_conv_path()
874 output_width_positions, mr, nr, deconvolution_op->ukernel.igemm.igemm_cases); in setup_subconv2d_path()
985 .ukernel = deconvolution_op->ukernel.igemm.igemm_cases[mr - 1], in setup_subconv2d_path()