Home
last modified time | relevance | path

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

/external/XNNPACK/src/
Dpacking.c25 float* packed_w, in xnn_pack_f32_gemm_goi_w() argument
36 packed_w[nr_block_offset] = b[nr_block_start + nr_block_offset]; in xnn_pack_f32_gemm_goi_w()
39 packed_w += nr; in xnn_pack_f32_gemm_goi_w()
44 *packed_w++ = in xnn_pack_f32_gemm_goi_w()
48 packed_w += (nr - nr_block_size) * kr; in xnn_pack_f32_gemm_goi_w()
55 *packed_w++ = in xnn_pack_f32_gemm_goi_w()
58 packed_w += kr - kr_block_size; in xnn_pack_f32_gemm_goi_w()
60 packed_w += (nr - nr_block_size) * kr; in xnn_pack_f32_gemm_goi_w()
79 uint16_t* packed_w, in xnn_pack_f16_gemm_goi_w() argument
90 packed_w[nr_block_offset] = b[nr_block_start + nr_block_offset]; in xnn_pack_f16_gemm_goi_w()
[all …]
Doperator-run.c42 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_grouped_gemm()
65 (const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride), in xnn_compute_gemm()
108 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_grouped_batch_igemm()
134 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_grouped_igemm()
160 (const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride), in xnn_compute_batch_igemm()
185 (const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride), in xnn_compute_igemm()
918 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_hmp_grouped_gemm()
942 (const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride), in xnn_compute_hmp_gemm()
968 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_hmp_grouped_batch_igemm()
995 …(const void*) ((uintptr_t) context->packed_w + nr_block_start * context->w_stride + group_index * … in xnn_compute_hmp_grouped_igemm()
[all …]
/external/XNNPACK/src/xnnpack/
Dpack.h42 void* packed_w,
54 float* packed_w,
66 uint16_t* packed_w,
78 void* packed_w,
90 void* packed_w,
102 void* packed_w,
114 void* packed_w,
125 float* packed_w,
136 uint16_t* packed_w,
147 void* packed_w,
[all …]
Dcompute.h68 const void* packed_w; member
169 const void* packed_w; member
/external/XNNPACK/test/
Dvmulcaddc-microkernel-tester.h134 std::vector<uint16_t, AlignedAllocator<uint16_t, 64>> packed_w(packed_channels() * 2);
149 std::fill(packed_w.begin(), packed_w.end(), UINT16_C(0x7E00) /* NaN */);
151 scale.data(), bias.data(), packed_w.data(), nullptr);
177 packed_w.data(),
204 std::vector<float, AlignedAllocator<float, 64>> packed_w(packed_channels() * 2);
218 std::fill(packed_w.begin(), packed_w.end(), nanf(""));
220 scale.data(), bias.data(), packed_w.data(), nullptr);
251 packed_w.data(),
Dgemm-microkernel-tester.h226 …std::vector<uint8_t, AlignedAllocator<uint8_t, 64>> packed_w(packed_n() * packed_k() + bias_n() * …
241 std::fill(packed_w.begin(), packed_w.end(), b_zero_point());
244 b.data(), bias.data(), packed_w.data(), &packing_params);
284 packed_w.data(),
319 …std::vector<uint8_t, AlignedAllocator<uint8_t, 64>> packed_w(ks() * packed_n() * packed_k() + bias…
339 std::fill(packed_w.begin(), packed_w.end(), b_zero_point());
343 b.data(), bias.data(), packed_w.data(), &packing_params);
410 im2col.data(), packed_w.data(),
448 …std::vector<int8_t, AlignedAllocator<int8_t, 64>> packed_w(packed_n() * packed_k() + bias_n() * si…
463 std::fill(packed_w.begin(), packed_w.end(), 0);
[all …]
/external/XNNPACK/src/operators/
Dfully-connected-nc.c210 .packed_w = fully_connected_op->packed_weights, in setup_fully_connected_nc()
Dconvolution-nhwc.c869 .packed_w = convolution_op->packed_weights, in setup_convolution2d_nhwc()
980 .packed_w = convolution_op->packed_weights, in setup_convolution2d_nhwc()
Ddeconvolution-nhwc.c532 .packed_w = deconvolution_op->packed_weights, in setup_conv_path()