Home
last modified time | relevance | path

Searched refs:subconvolution_params (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/
H A Doperator-run.c477 …const struct subconvolution_params* subconvolution_params = &context->subconvolution_params[subker… in xnn_compute_grouped_subgemm2d() local
479 if XNN_UNLIKELY(slice_y >= subconvolution_params->slice_height) { in xnn_compute_grouped_subgemm2d()
483 const size_t slice_width = subconvolution_params->slice_width; in xnn_compute_grouped_subgemm2d()
497 …(const void*) ((uintptr_t) subconvolution_params->weights + nc_block_start * subconvolution_params… in xnn_compute_grouped_subgemm2d()
498 …(void*) ((uintptr_t) subconvolution_params->output + group_index * context->gc_stride + slice_y * … in xnn_compute_grouped_subgemm2d()
514 …const struct subconvolution_params* subconvolution_params = &context->subconvolution_params[subker… in xnn_compute_subgemm2d() local
516 if XNN_UNLIKELY(slice_y >= subconvolution_params->slice_height) { in xnn_compute_subgemm2d()
520 const size_t slice_width = subconvolution_params->slice_width; in xnn_compute_subgemm2d()
534 …(const void*) ((uintptr_t) subconvolution_params->weights + nc_block_start * subconvolution_params… in xnn_compute_subgemm2d()
535 …(void*) ((uintptr_t) subconvolution_params->output + slice_y * context->cy_stride + slice_x_start … in xnn_compute_subgemm2d()
[all …]
H A Dindirection.c142 struct subconvolution_params* subconvolution_params = op->subconvolution_buffer; in xnn_indirection_init_subconv2d() local
165 subconvolution_params->indirection_buffer = indirection_buffer; in xnn_indirection_init_subconv2d()
166 subconvolution_params->indirection_y_stride = in xnn_indirection_init_subconv2d()
167 … subconvolution_params->indirection_x_stride * round_up(sliced_output_width, output_tile_size); in xnn_indirection_init_subconv2d()
168 ++subconvolution_params; in xnn_indirection_init_subconv2d()
H A Dpacking.c1084 struct subconvolution_params* subconv_params, in xnn_pack_f32_deconv_goki_w()
1144 struct subconvolution_params* subconv_params, in xnn_pack_f16_deconv_goki_w()
1204 struct subconvolution_params* subconv_params, in xnn_pack_f32_to_f16_deconv_goki_w()
1264 struct subconvolution_params* subconv_params, in xnn_pack_qs8_deconv_goki_w()
1337 struct subconvolution_params* subconv_params, in xnn_pack_qu8_deconv_goki_w()
/aosp_15_r20/external/XNNPACK/src/operators/
H A Ddeconvolution-nhwc.c169 const size_t subconvolution_buffer_size = sizeof(struct subconvolution_params) * subkernels; in create_deconvolution2d_nhwc()
178 struct subconvolution_params* subconvolution_params = deconvolution_op->subconvolution_buffer; in create_deconvolution2d_nhwc() local
185 subconvolution_params->indirection_x_stride = sizeof(void*) * subkernel_size; in create_deconvolution2d_nhwc()
186 …subconvolution_params->w_stride = bias_element_size + ((k_stride * subkernel_size) << log2_filter_… in create_deconvolution2d_nhwc()
187 subconvolution_params++; in create_deconvolution2d_nhwc()
888 struct subconvolution_params* subconvolution_params = deconvolution_op->subconvolution_buffer; in setup_subconv2d_path() local
889 if (packed_weights_ptr != subconvolution_params->weights) { in setup_subconv2d_path()
891 …const ptrdiff_t diff = (uintptr_t) packed_weights_ptr - (uintptr_t) subconvolution_params->weights; in setup_subconv2d_path()
894 … subconvolution_params->weights = (void*) ((uintptr_t) subconvolution_params->weights + diff); in setup_subconv2d_path()
895 ++subconvolution_params; in setup_subconv2d_path()
[all …]
/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A Dpack.h370 struct subconvolution_params* subconv_params,
387 struct subconvolution_params* subconv_params,
404 struct subconvolution_params* subconv_params,
421 struct subconvolution_params* subconv_params,
438 struct subconvolution_params* subconv_params,
455 struct subconvolution_params* subconv_params,
H A Dcompute.h367 const struct subconvolution_params* subconvolution_params; member
414 const struct subconvolution_params* subconvolution_params; member
H A Doperator.h108 struct subconvolution_params { struct
185 struct subconvolution_params* subconvolution_buffer;