Searched refs:subconvolution_params (Results 1 – 7 of 7) sorted by relevance
| /external/XNNPACK/src/ |
| D | operator-run.c | 205 …const struct subconvolution_params* subconvolution_params = &context->subconvolution_params[subker… in xnn_compute_grouped_subgemm2d() local
207 if XNN_UNLIKELY(slice_y >= subconvolution_params->slice_height) { in xnn_compute_grouped_subgemm2d()
211 const size_t slice_width = subconvolution_params->slice_width; in xnn_compute_grouped_subgemm2d()
225 …(const void*) ((uintptr_t) subconvolution_params->weights + nc_block_start * subconvolution_params… in xnn_compute_grouped_subgemm2d()
226 …(void*) ((uintptr_t) subconvolution_params->output + group_index * context->gc_stride + slice_y * … in xnn_compute_grouped_subgemm2d()
242 …const struct subconvolution_params* subconvolution_params = &context->subconvolution_params[subker… in xnn_compute_subgemm2d() local
244 if XNN_UNLIKELY(slice_y >= subconvolution_params->slice_height) { in xnn_compute_subgemm2d()
248 const size_t slice_width = subconvolution_params->slice_width; in xnn_compute_subgemm2d()
262 …(const void*) ((uintptr_t) subconvolution_params->weights + nc_block_start * subconvolution_params… in xnn_compute_subgemm2d()
263 …(void*) ((uintptr_t) subconvolution_params->output + slice_y * context->cy_stride + slice_x_start … in xnn_compute_subgemm2d()
[all …]
|
| D | indirection.c | 139 struct subconvolution_params* subconvolution_params = op->subconvolution_buffer; in xnn_indirection_init_subconv2d() local
162 subconvolution_params->indirection_buffer = indirection_buffer; in xnn_indirection_init_subconv2d()
163 subconvolution_params->indirection_y_stride = in xnn_indirection_init_subconv2d()
164 … subconvolution_params->indirection_x_stride * round_up(sliced_output_width, output_tile_size); in xnn_indirection_init_subconv2d()
165 ++subconvolution_params; in xnn_indirection_init_subconv2d()
|
| D | packing.c | 841 struct subconvolution_params* subconv_params, in xnn_pack_f32_deconv_goki_w()
910 struct subconvolution_params* subconv_params, in xnn_pack_f16_deconv_goki_w()
979 struct subconvolution_params* subconv_params, in xnn_pack_qu8_deconv_goki_w()
|
| /external/XNNPACK/src/operators/ |
| D | deconvolution-nhwc.c | 184 const size_t subconvolution_buffer_size = sizeof(struct subconvolution_params) * subkernels; in create_deconvolution2d_nhwc()
193 struct subconvolution_params* subconvolution_params = deconvolution_op->subconvolution_buffer; in create_deconvolution2d_nhwc() local
200 subconvolution_params->indirection_x_stride = sizeof(void*) * subkernel_size; in create_deconvolution2d_nhwc()
201 subconvolution_params->w_stride = sizeof(float) + k_stride * subkernel_size * sizeof(float); in create_deconvolution2d_nhwc()
202 subconvolution_params++; in create_deconvolution2d_nhwc()
636 struct subconvolution_params* subconvolution_params = deconvolution_op->subconvolution_buffer; in setup_subconv2d_path() local
643 … subconvolution_params->scaled_kernel_size = mr * subconvolution_params->indirection_x_stride; in setup_subconv2d_path()
644 … subconvolution_params->slice_width = divide_round_up(output_width - output_x_start, stride_width); in setup_subconv2d_path()
645 …subconvolution_params->slice_height = divide_round_up(output_height - output_y_start, stride_heigh… in setup_subconv2d_path()
646 subconvolution_params->output = in setup_subconv2d_path()
[all …]
|
| /external/XNNPACK/src/xnnpack/ |
| D | pack.h | 287 struct subconvolution_params* subconv_params,
304 struct subconvolution_params* subconv_params,
321 struct subconvolution_params* subconv_params,
338 struct subconvolution_params* subconv_params,
|
| D | operator.h | 179 struct subconvolution_params { struct
256 struct subconvolution_params* subconvolution_buffer;
|
| D | compute.h | 261 const struct subconvolution_params* subconvolution_params; member
308 const struct subconvolution_params* subconvolution_params; member
|