Home
last modified time | relevance | path

Searched refs:convolution_op (Results 1 – 9 of 9) sorted by relevance

/external/XNNPACK/src/operators/
Dconvolution-nhwc.c104 xnn_operator_t convolution_op = NULL; in create_convolution2d_nhwc() local
209 convolution_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in create_convolution2d_nhwc()
210 if (convolution_op == NULL) { in create_convolution2d_nhwc()
243 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in create_convolution2d_nhwc()
244 if (convolution_op->packed_weights == NULL) { in create_convolution2d_nhwc()
253 kernel, bias, convolution_op->packed_weights, packing_params); in create_convolution2d_nhwc()
255 convolution_op->ukernel.vmulcaddc = (struct xnn_ukernel_vmulcaddc) { in create_convolution2d_nhwc()
268 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in create_convolution2d_nhwc()
269 if (convolution_op->packed_weights == NULL) { in create_convolution2d_nhwc()
275 memset(convolution_op->packed_weights, packed_weights_padding_byte, packed_weights_size); in create_convolution2d_nhwc()
[all …]
Dconvolution-nchw.c60 xnn_operator_t convolution_op = NULL; in xnn_create_convolution2d_nchw_f32() local
223 convolution_op = xnn_allocate_zero_simd_memory(sizeof(struct xnn_operator)); in xnn_create_convolution2d_nchw_f32()
224 if (convolution_op == NULL) { in xnn_create_convolution2d_nchw_f32()
305 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_convolution2d_nchw_f32()
306 if (convolution_op->packed_weights == NULL) { in xnn_create_convolution2d_nchw_f32()
312 convolution_op->num_nonzero_values = num_nonzero_values; in xnn_create_convolution2d_nchw_f32()
313 convolution_op->num_nonzero_blocks = num_nonzero_blocks; in xnn_create_convolution2d_nchw_f32()
314 convolution_op->num_output_channel_blocks = num_output_channel_blocks; in xnn_create_convolution2d_nchw_f32()
316 float* nonzero_values = convolution_op->packed_weights; in xnn_create_convolution2d_nchw_f32()
401 convolution_op->first_input_channel = first_ic; in xnn_create_convolution2d_nchw_f32()
[all …]
/external/XNNPACK/bench/
Df16-igemm.cc97 xnn_operator convolution_op = { }; in IGEMMBenchmark() local
98 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in IGEMMBenchmark()
99 convolution_op.input = a.data(); in IGEMMBenchmark()
100 convolution_op.input_pixel_stride = input_pixel_stride; in IGEMMBenchmark()
101 convolution_op.zero_buffer = z.data(); in IGEMMBenchmark()
102 convolution_op.groups = 1; in IGEMMBenchmark()
103 convolution_op.group_input_channels = group_input_channels; in IGEMMBenchmark()
104 convolution_op.batch_size = 1; in IGEMMBenchmark()
105 convolution_op.input_height = input_height; in IGEMMBenchmark()
106 convolution_op.input_width = input_width; in IGEMMBenchmark()
[all …]
Df16-dwconv.cc100 xnn_operator convolution_op = { }; in DWConvBenchmark() local
101 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in DWConvBenchmark()
102 convolution_op.input = a.data(); in DWConvBenchmark()
103 convolution_op.input_pixel_stride = channels; in DWConvBenchmark()
104 convolution_op.zero_buffer = z.data(); in DWConvBenchmark()
105 convolution_op.input_height = input_height; in DWConvBenchmark()
106 convolution_op.input_width = input_width; in DWConvBenchmark()
107 convolution_op.output_height = output_height; in DWConvBenchmark()
108 convolution_op.output_width = output_width; in DWConvBenchmark()
109 convolution_op.kernel_height = kernel_height; in DWConvBenchmark()
[all …]
Df32-dwconv.cc92 xnn_operator convolution_op = { }; in DWConvBenchmark() local
93 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in DWConvBenchmark()
94 convolution_op.input = a.data(); in DWConvBenchmark()
95 convolution_op.input_pixel_stride = channels; in DWConvBenchmark()
96 convolution_op.zero_buffer = z.data(); in DWConvBenchmark()
97 convolution_op.input_height = input_height; in DWConvBenchmark()
98 convolution_op.input_width = input_width; in DWConvBenchmark()
99 convolution_op.output_height = output_height; in DWConvBenchmark()
100 convolution_op.output_width = output_width; in DWConvBenchmark()
101 convolution_op.kernel_height = kernel_height; in DWConvBenchmark()
[all …]
Df32-igemm.cc91 xnn_operator convolution_op = { }; in IGEMMBenchmark() local
92 convolution_op.indirection_buffer = reinterpret_cast<const void**>(i.data()); in IGEMMBenchmark()
93 convolution_op.input = a.data(); in IGEMMBenchmark()
94 convolution_op.input_pixel_stride = input_pixel_stride; in IGEMMBenchmark()
95 convolution_op.zero_buffer = z.data(); in IGEMMBenchmark()
96 convolution_op.groups = 1; in IGEMMBenchmark()
97 convolution_op.group_input_channels = group_input_channels; in IGEMMBenchmark()
98 convolution_op.batch_size = 1; in IGEMMBenchmark()
99 convolution_op.input_height = input_height; in IGEMMBenchmark()
100 convolution_op.input_width = input_width; in IGEMMBenchmark()
[all …]
Dconvolution.cc91 for (xnn_operator_t& convolution_op : convolution_operators) { in xnnpack_convolution_qu8()
103 0 /* flags */, &convolution_op); in xnnpack_convolution_qu8()
137 for (xnn_operator_t& convolution_op : convolution_operators) { in xnnpack_convolution_qu8()
138 status = xnn_delete_operator(convolution_op); in xnnpack_convolution_qu8()
143 convolution_op = nullptr; in xnnpack_convolution_qu8()
212 for (xnn_operator_t& convolution_op : convolution_operators) { in xnnpack_convolution_qs8()
223 0 /* flags */, &convolution_op); in xnnpack_convolution_qs8()
257 for (xnn_operator_t& convolution_op : convolution_operators) { in xnnpack_convolution_qs8()
258 status = xnn_delete_operator(convolution_op); in xnnpack_convolution_qs8()
263 convolution_op = nullptr; in xnnpack_convolution_qs8()
[all …]
/external/XNNPACK/test/
Dconvolution-operator-tester.h620 xnn_operator_t convolution_op = nullptr; in TestNHWCxQS8() local
634 &convolution_op); in TestNHWCxQS8()
639 ASSERT_NE(nullptr, convolution_op); in TestNHWCxQS8()
642 …tr<xnn_operator, decltype(&xnn_delete_operator)> auto_convolution_op(convolution_op, xnn_delete_op… in TestNHWCxQS8()
646 convolution_op, in TestNHWCxQS8()
652 xnn_run_operator(convolution_op, nullptr /* thread pool */)); in TestNHWCxQS8()
788 xnn_operator_t convolution_op = nullptr; in TestNHWCxQU8() local
803 &convolution_op); in TestNHWCxQU8()
808 ASSERT_NE(nullptr, convolution_op); in TestNHWCxQU8()
811 …tr<xnn_operator, decltype(&xnn_delete_operator)> auto_convolution_op(convolution_op, xnn_delete_op… in TestNHWCxQU8()
[all …]
/external/XNNPACK/include/
Dxnnpack.h1250 xnn_operator_t convolution_op,
1671 xnn_operator_t convolution_op,
1846 xnn_operator_t convolution_op,
1958 xnn_operator_t convolution_op,
2051 xnn_operator_t convolution_op,