Searched defs:k_stride (Results 1 – 15 of 15) sorted by relevance
14 size_t k_stride; member40 const size_t k_stride = context->k_stride; in compute_q8gemm_xzp() local69 size_t k_stride; member91 const size_t k_stride = context->k_stride; in compute_q8gemm() local526 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in qnnpackConv() local603 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in qnnpackConv() local640 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in qnnpackConv() local
8 size_t k_stride; member32 const size_t k_stride = context->k_stride; in compute_q8gemm() local78 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in qnnpackLinear() local
8 size_t k_stride; member32 const size_t k_stride = context->k_stride; in compute_q8gemm_dq() local78 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in qnnpackLinearDynamic() local
166 const uint32_t k_stride = in PrePackConvWeights() local203 const uint32_t k_stride = in PrePackConvWeights() local
27 size_t k_stride; member50 const size_t k_stride = context->k_stride; in compute_q8gemm() local296 size_t k_stride; member323 const size_t k_stride = context->k_stride; in compute_q8gemm_xzp() local1009 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_run_operator() local1077 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_run_operator() local1180 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_run_operator() local1288 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_run_operator() local
376 const uint32_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_create_convolution_ndhwc_q8() local414 const uint32_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_create_convolution_ndhwc_q8() local
34 const uint32_t k_stride = (input_channels + (kr - 1)) & -kr; in PackBMatrix() local
72 const uint32_t k_stride = (input_channels + (kr - 1)) & -kr; in pytorch_qnnp_create_fully_connected_nc_q8() local
129 const uint32_t k_stride = (group_input_channels + (kr - 1)) & -kr; in pytorch_qnnp_create_deconvolution2d_nhwc_q8() local
119 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in qnnpackDeConv() local
235 const size_t k_stride = (group_input_channels + (kr - 1)) & -kr; in PackedConvWeightsQnnp() local
119 const size_t k_stride = round_up_po2(input_channels, kr * sr); in create_fully_connected_nc() local
159 const uint32_t k_stride = round_up_po2(group_input_channels, kr * sr); in create_deconvolution2d_nhwc() local
497 const size_t k_stride = round_up_po2(group_input_channels, kr * sr); in create_convolution2d_nhwc() local
127 std::array<int, MAX_MHA_DIM> k_stride; member