| /external/XNNPACK/src/ |
| D | packing.c | 30 float* packed_w, in xnn_pack_f32_gemm_goi_w()
77 uint16_t* packed_w, in xnn_pack_f16_gemm_goi_w()
124 uint16_t* packed_w, in xnn_pack_f32_to_f16_gemm_goi_w()
171 void* packed_w, in xnn_pack_qu8_gemm_goi_w()
232 void* packed_w, in xnn_pack_qs8_gemm_goi_w()
292 void* packed_w, in xnn_pack_qs8_gemm_xw_goi_w()
351 float* packed_w, in xnn_pack_f32_gemm_io_w()
389 uint16_t* packed_w, in xnn_pack_f16_gemm_io_w()
427 uint16_t* packed_w, in xnn_pack_f32_to_f16_gemm_io_w()
465 void* packed_w, in xnn_pack_qu8_gemm_io_w()
[all …]
|
| D | microparams-init.c | 757 void* packed_w) in xnn_init_qc8_scale_fp32_params()
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/qnnpack/ |
| D | pack.h | 27 void* packed_w) { in pytorch_pack_q8gemm_wdq()
76 void* const packed_w) { in pytorch_pack_q8gemm_wrq()
149 void* packed_w) { in pytorch_pack_q8conv_wdq()
197 void* const packed_w) { in pytorch_pack_q8conv_wrq()
271 void* packed_w) { in pytorch_pack_q8deconv_wdq()
319 void* const packed_w) { in pytorch_pack_q8deconv_wrq()
392 void* packed_w) { in pytorch_pack_q8dw_wdq()
431 void* const packed_w) { in pytorch_pack_q8dw_wrq()
473 void* packed_w, in pytorch_pack_q8dw_3d_w_dilation()
514 void* packed_w, in pytorch_pack_q8dw_2d_w_dilation()
[all …]
|
| /external/XNNPACK/test/ |
| D | gemm-microkernel-tester.cc | 42 …std::vector<uint8_t, AlignedAllocator<uint8_t, 64>> packed_w(packed_n() * packed_k() + packed_n() … in Test() local
131 …std::vector<uint8_t, AlignedAllocator<uint8_t, 64>> packed_w(ks() * packed_n() * packed_k() + pack… in Test() local
258 …std::vector<int8_t, AlignedAllocator<int8_t, 64>> packed_w(packed_n() * packed_k() + packed_n() * … in Test() local
375 …std::vector<int8_t, AlignedAllocator<int8_t, 64>> packed_w(ks() * packed_n() * packed_k() + packed… in Test() local
513 …std::vector<int8_t, AlignedAllocator<int8_t, 64>> packed_w(packed_n() * packed_k() + packed_n() * … in Test() local
612 …std::vector<int8_t, AlignedAllocator<int8_t, 64>> packed_w(ks() * packed_n() * packed_k() + packed… in Test() local
731 …std::vector<uint16_t, AlignedAllocator<uint16_t, 64>> packed_w(packed_n() * packed_k() + packed_n(… in Test() local
808 …std::vector<uint16_t, AlignedAllocator<uint16_t, 64>> packed_w(packed_n() * packed_k() + packed_n(… in Test() local
880 …std::vector<uint16_t, AlignedAllocator<uint16_t, 64>> packed_w(ks() * packed_k() * packed_n() + pa… in Test() local
1000 std::vector<float, AlignedAllocator<float, 64>> packed_w(packed_n() * packed_k() + packed_n()); in Test() local
[all …]
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| D | fc-run.cc | 13 const uint8_t* packed_w; member
37 const void* packed_w = context->packed_w; in compute_q8gemm() local
|
| D | fc-dynamic-run.cc | 13 const uint8_t* packed_w; member
37 const void* packed_w = context->packed_w; in compute_q8gemm_dq() local
|
| D | conv-run.cc | 19 const void* packed_w; member
45 const void* packed_w = context->packed_w; in compute_q8gemm_xzp() local
74 const uint8_t* packed_w; member
96 const void* packed_w = context->packed_w; in compute_q8gemm() local
125 const void* packed_w; member
150 const void* packed_w = context->packed_w; in compute_q8conv() local
|
| D | deconv-run.cc | 37 const void* packed_w; member
63 const void* packed_w = context->packed_w; in compute_q8conv() local
|
| D | operator-run.c | 32 const uint8_t* packed_w; member
55 const void* restrict packed_w = context->packed_w; in compute_q8gemm() local
301 const void* packed_w; member
328 const void* restrict packed_w = context->packed_w; in compute_q8gemm_xzp() local
360 const void* packed_w; member
386 const void* restrict packed_w = context->packed_w; in compute_q8conv() local
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| D | fbgemm_utils.h | 377 at::Tensor packed_w; member
|
| /external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/q8gemm_sparse/ |
| D | 8x4c1x4-dq-packedA-sse2.h | 36 const uint8_t* packed_w, in KERNEL_NAME()
|
| /external/pytorch/aten/src/ATen/native/mkldnn/ |
| D | MKLDNNConversions.cpp | 523 const ideep::tensor packed_w = itensor_from_tensor(self); in get_mkldnn_serialized_md() local
|
| /external/pytorch/torch/_inductor/ |
| D | mkldnn_ir.py | 1126 def create(cls, x, packed_w, orig_w, B, batch_size): argument
|
| D | ir.py | 6279 def create(cls, x, packed_w, orig_w, B, batch_size): argument
|
| /external/XNNPACK/src/xnnpack/ |
| D | compute.h | 173 const void* packed_w; member
275 const void* packed_w; member
|