Home
last modified time | relevance | path

Searched refs:packed_weights (Results 1 – 23 of 23) sorted by relevance

/external/XNNPACK/test/
Dibilinear-microkernel-tester.h101 std::vector<float, AlignedAllocator<float, 64>> packed_weights(pixels() * 2); in Test()
107 std::generate(packed_weights.begin(), packed_weights.end(), std::ref(f32rng)); in Test()
118 const float alpha_h = packed_weights[i * 2 + 0]; in Test()
119 const float alpha_v = packed_weights[i * 2 + 1]; in Test()
132 packed_weights.data(), output.data(), in Test()
155 std::vector<float, AlignedAllocator<float, 64>> packed_weights(pixels() * 2); in TestCHW()
161 std::generate(packed_weights.begin(), packed_weights.end(), std::ref(f32rng)); in TestCHW()
175 const float alpha_h = packed_weights[i * 2 + 0]; in TestCHW()
176 const float alpha_v = packed_weights[i * 2 + 1]; in TestCHW()
191 packed_weights.data(), output.data(), input_stride() * sizeof(float)); in TestCHW()
Ddwconv-microkernel-tester.h182 …std::vector<uint8_t, AlignedAllocator<uint8_t, 64>> packed_weights((kr() + sizeof(int32_t) / sizeo…
199 std::fill(packed_weights.begin(), packed_weights.end(), 0);
203 kernel.data(), bias.data(), packed_weights.data(), &packing_params);
264 indirection.data(), packed_weights.data(), output.data(),
295 …std::vector<int8_t, AlignedAllocator<int8_t, 64>> packed_weights((kr() + sizeof(int32_t) / sizeof(…
312 std::fill(packed_weights.begin(), packed_weights.end(), 0);
316 kernel.data(), bias.data(), packed_weights.data(), &packing_params);
377 indirection.data(), packed_weights.data(), output.data(),
407 …std::vector<uint16_t, AlignedAllocator<uint16_t, 64>> packed_weights((kr() + 1) * packed_channels(…
420 std::fill(packed_weights.begin(), packed_weights.end(), 0);
[all …]
Ddwconv2d-microkernel-tester.h179 std::vector<float> packed_weights(kernel_size() + 1);
185 std::generate(packed_weights.begin(), packed_weights.end(), std::ref(f32rng));
190 float acc = packed_weights[0];
197 const float kernel_val = packed_weights[1 + ky * kernel_width() + kx];
232 input.data(), packed_weights.data(), zero.data(), output.data(),
Dconv-hwc2chw-microkernel-tester.h308 …std::vector<float, AlignedAllocator<float, 64>> packed_weights((input_channels() * kernel_height()…
315 std::fill(packed_weights.begin(), packed_weights.end(), 0.0f);
322 kernel.data(), bias.data(), packed_weights.data(), nullptr);
378 input.data(), zero.data(), packed_weights.data(), output.data(),
Dconv-hwc-microkernel-tester.h302 …std::vector<float, AlignedAllocator<float, 64>> packed_weights((input_channels() * kernel_height()…
309 std::fill(packed_weights.begin(), packed_weights.end(), 0.0f);
316 kernel.data(), bias.data(), packed_weights.data(), nullptr);
372 input.data(), zero.data(), packed_weights.data(), output.data(),
/external/XNNPACK/bench/
Df32-conv-hwc2chw.cc67 std::vector<float, AlignedAllocator<float, 32>> packed_weights(weights_elements * num_buffers); in DConvHWC2CHW3X3S2P1Benchmark() local
68 std::fill(packed_weights.begin(), packed_weights.end(), 0.0f); in DConvHWC2CHW3X3S2P1Benchmark()
72 kernel.data(), bias.data(), packed_weights.data(), NULL); in DConvHWC2CHW3X3S2P1Benchmark()
74 std::copy(packed_weights.cbegin(), in DConvHWC2CHW3X3S2P1Benchmark()
75 packed_weights.cbegin() + weights_elements, in DConvHWC2CHW3X3S2P1Benchmark()
76 packed_weights.begin() + n * weights_elements); in DConvHWC2CHW3X3S2P1Benchmark()
96 packed_weights.data() + buffer_index * weights_elements, in DConvHWC2CHW3X3S2P1Benchmark()
Df32-conv-hwc.cc65 std::vector<float, AlignedAllocator<float, 32>> packed_weights(weights_elements * num_buffers); in DConv3X3S2P1Benchmark() local
66 std::fill(packed_weights.begin(), packed_weights.end(), 0.0f); in DConv3X3S2P1Benchmark()
70 kernel.data(), bias.data(), packed_weights.data(), nullptr); in DConv3X3S2P1Benchmark()
72 std::copy(packed_weights.cbegin(), in DConv3X3S2P1Benchmark()
73 packed_weights.cbegin() + weights_elements, in DConv3X3S2P1Benchmark()
74 packed_weights.begin() + n * weights_elements); in DConv3X3S2P1Benchmark()
94 packed_weights.data() + buffer_index * weights_elements, in DConv3X3S2P1Benchmark()
Df32-dwconv2d-chw.cc97 std::vector<float, AlignedAllocator<float, 32>> packed_weights(w_elements * num_buffers); in DWConv2DBenchmark() local
98 std::fill(packed_weights.begin(), packed_weights.end(), 0.0f); in DWConv2DBenchmark()
100 packed_weights[c * kernel_size + c] = bias[c]; in DWConv2DBenchmark()
102 packed_weights[c * kernel_size + c + 1 + i] = kernel[c * kernel_size + i]; in DWConv2DBenchmark()
106 …std::copy(packed_weights.cbegin(), packed_weights.cbegin() + w_elements, packed_weights.begin() + … in DWConv2DBenchmark()
126 packed_weights.data() + channel * (kernel_size + 1) + buffer_index * w_elements, in DWConv2DBenchmark()
/external/XNNPACK/src/operators/
Dresize-bilinear-nhwc.c163 xnn_release_simd_memory(resize_op->packed_weights); in xnn_setup_resize_bilinear2d_nhwc_f32()
164 resize_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_setup_resize_bilinear2d_nhwc_f32()
165 if (resize_op->packed_weights == NULL) { in xnn_setup_resize_bilinear2d_nhwc_f32()
184 input, resize_op->indirection_buffer, resize_op->packed_weights, in xnn_setup_resize_bilinear2d_nhwc_f32()
201 .packed_weights = resize_op->packed_weights, in xnn_setup_resize_bilinear2d_nhwc_f32()
Dresize-bilinear-nchw.c163 xnn_release_simd_memory(resize_op->packed_weights); in xnn_setup_resize_bilinear2d_nchw_f32()
164 resize_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_setup_resize_bilinear2d_nchw_f32()
165 if (resize_op->packed_weights == NULL) { in xnn_setup_resize_bilinear2d_nchw_f32()
184 input, resize_op->indirection_buffer, resize_op->packed_weights, in xnn_setup_resize_bilinear2d_nchw_f32()
202 .packed_weights = resize_op->packed_weights, in xnn_setup_resize_bilinear2d_nchw_f32()
Dconvolution-nchw.c305 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()
316 float* nonzero_values = convolution_op->packed_weights; in xnn_create_convolution2d_nchw_f32()
418 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_convolution2d_nchw_f32()
419 if (convolution_op->packed_weights == NULL) { in xnn_create_convolution2d_nchw_f32()
431 kernel, bias, convolution_op->packed_weights, NULL); in xnn_create_convolution2d_nchw_f32()
448 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_convolution2d_nchw_f32()
449 if (convolution_op->packed_weights == NULL) { in xnn_create_convolution2d_nchw_f32()
459 kernel, bias, convolution_op->packed_weights, NULL); in xnn_create_convolution2d_nchw_f32()
463 kernel, bias, convolution_op->packed_weights, NULL); in xnn_create_convolution2d_nchw_f32()
[all …]
Dprelu-nc.c73 prelu_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in xnn_create_prelu_nc_f32()
74 if (prelu_op->packed_weights == NULL) { in xnn_create_prelu_nc_f32()
80 memcpy(prelu_op->packed_weights, negative_slope, channels * sizeof(float)); in xnn_create_prelu_nc_f32()
129 .w = prelu_op->packed_weights, in xnn_setup_prelu_nc_f32()
Dconvolution-nhwc.c243 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()
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()
281 kernel, bias, convolution_op->packed_weights, packing_params); in create_convolution2d_nhwc()
286 kernel, bias, convolution_op->packed_weights, packing_params); in create_convolution2d_nhwc()
312 convolution_op->packed_weights = xnn_allocate_simd_memory(packed_group_weights_size * groups); in create_convolution2d_nhwc()
313 if (convolution_op->packed_weights == NULL) { in create_convolution2d_nhwc()
[all …]
Dfully-connected-nc.c106 fully_connected_op->packed_weights = xnn_allocate_simd_memory(packed_weights_size); in create_fully_connected_nc()
107 if (fully_connected_op->packed_weights == NULL) { in create_fully_connected_nc()
113 memset(fully_connected_op->packed_weights, packed_weights_padding_byte, packed_weights_size); in create_fully_connected_nc()
120 fully_connected_op->packed_weights, in create_fully_connected_nc()
127 fully_connected_op->packed_weights, in create_fully_connected_nc()
210 .packed_w = fully_connected_op->packed_weights, in setup_fully_connected_nc()
Ddeconvolution-nhwc.c206 deconvolution_op->packed_weights = xnn_allocate_simd_memory(packed_group_weights_size * groups); in create_deconvolution2d_nhwc()
207 if (deconvolution_op->packed_weights == NULL) { in create_deconvolution2d_nhwc()
213 …memset(deconvolution_op->packed_weights, packed_weights_padding_byte, packed_group_weights_size * … in create_deconvolution2d_nhwc()
220 kernel, bias, deconvolution_op->packed_weights, in create_deconvolution2d_nhwc()
228 kernel, bias, deconvolution_op->packed_weights, deconvolution_op->subconvolution_buffer, in create_deconvolution2d_nhwc()
532 .packed_w = deconvolution_op->packed_weights, in setup_conv_path()
/external/XNNPACK/src/
Dindirection.c335 float* packed_weights, in xnn_indirection_init_resize_bilinear2d_hwc_f32() argument
382 packed_weights[0] = alpha_x; in xnn_indirection_init_resize_bilinear2d_hwc_f32()
383 packed_weights[1] = alpha_y; in xnn_indirection_init_resize_bilinear2d_hwc_f32()
385 packed_weights += 2; in xnn_indirection_init_resize_bilinear2d_hwc_f32()
413 packed_weights[0] = alpha_x; in xnn_indirection_init_resize_bilinear2d_hwc_f32()
414 packed_weights[1] = alpha_y; in xnn_indirection_init_resize_bilinear2d_hwc_f32()
416 packed_weights += 2; in xnn_indirection_init_resize_bilinear2d_hwc_f32()
430 float* packed_weights, in xnn_indirection_init_resize_bilinear2d_chw_f32() argument
479 packed_weights[0] = alpha_x; in xnn_indirection_init_resize_bilinear2d_chw_f32()
480 packed_weights[1] = alpha_y; in xnn_indirection_init_resize_bilinear2d_chw_f32()
[all …]
Doperator-delete.c30 xnn_release_simd_memory(op->packed_weights); in xnn_delete_operator()
Dpacking.c1515 float* packed_weights, in xnn_pack_f32_chw_dwconv_ghw_w() argument
1520 *packed_weights = *bias++; in xnn_pack_f32_chw_dwconv_ghw_w()
1522 *packed_weights = 0.0f; in xnn_pack_f32_chw_dwconv_ghw_w()
1524 packed_weights += 1; in xnn_pack_f32_chw_dwconv_ghw_w()
1526 *packed_weights++ = kernel[g * kernel_size + i]; in xnn_pack_f32_chw_dwconv_ghw_w()
1536 uint16_t* packed_weights, in xnn_pack_f16_chw_dwconv_ghw_w() argument
1541 *packed_weights = *bias++; in xnn_pack_f16_chw_dwconv_ghw_w()
1543 *packed_weights = 0; in xnn_pack_f16_chw_dwconv_ghw_w()
1545 packed_weights += 1; in xnn_pack_f16_chw_dwconv_ghw_w()
1547 *packed_weights++ = kernel[g * kernel_size + i]; in xnn_pack_f16_chw_dwconv_ghw_w()
[all …]
Doperator-run.c359 context->packed_weights, in xnn_compute_conv2d_hwc2chw()
381 indirect_input, context->packed_weights, output, in xnn_compute_dwconv_unipass()
396 (const void*) ((uintptr_t) context->packed_weights + channel * context->weights_channel_stride), in xnn_compute_dwconv2d_chw()
694 (const void*) ((uintptr_t) context->packed_weights + (pixel_start << context->log2_wsize)), in xnn_compute_resize_bilinear()
714 context->packed_weights, in xnn_compute_resize_bilinear_chw()
/external/XNNPACK/src/xnnpack/
Dindirection.h57 float* packed_weights,
69 float* packed_weights,
Dcompute.h359 const void* packed_weights; member
388 const void* packed_weights; member
422 const void* packed_weights; member
710 const void* packed_weights; member
737 const void* packed_weights; member
Dpack.h495 float* packed_weights,
503 uint16_t* packed_weights,
512 float* packed_weights,
Doperator.h226 void* packed_weights; member