/external/XNNPACK/test/ |
D | ibilinear-microkernel-tester.h | 101 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()
|
D | dwconv-microkernel-tester.h | 182 …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 …]
|
D | dwconv2d-microkernel-tester.h | 179 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(),
|
D | conv-hwc2chw-microkernel-tester.h | 308 …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(),
|
D | conv-hwc-microkernel-tester.h | 302 …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/ |
D | f32-conv-hwc2chw.cc | 67 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()
|
D | f32-conv-hwc.cc | 65 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()
|
D | f32-dwconv2d-chw.cc | 97 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/ |
D | resize-bilinear-nhwc.c | 163 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()
|
D | resize-bilinear-nchw.c | 163 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()
|
D | convolution-nchw.c | 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() 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 …]
|
D | prelu-nc.c | 73 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()
|
D | convolution-nhwc.c | 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() 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 …]
|
D | fully-connected-nc.c | 106 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()
|
D | deconvolution-nhwc.c | 206 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/ |
D | indirection.c | 335 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 …]
|
D | operator-delete.c | 30 xnn_release_simd_memory(op->packed_weights); in xnn_delete_operator()
|
D | packing.c | 1515 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 …]
|
D | operator-run.c | 359 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/ |
D | indirection.h | 57 float* packed_weights, 69 float* packed_weights,
|
D | compute.h | 359 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
|
D | pack.h | 495 float* packed_weights, 503 uint16_t* packed_weights, 512 float* packed_weights,
|
D | operator.h | 226 void* packed_weights; member
|