| /aosp_15_r20/external/XNNPACK/src/xnnpack/ |
| H A D | params.h | 29 struct xnn_hmp_gemm_ukernel ukernel = {{ function }}; in xnn_init_hmp_gemm_ukernel() local
39 static inline bool xnn_is_hmp_gemm_ukernel(struct xnn_hmp_gemm_ukernel ukernel) { in xnn_is_hmp_gemm_ukernel()
60 struct xnn_hmp_igemm_ukernel ukernel = {{ function }}; in xnn_init_hmp_igemm_ukernel() local
70 static inline bool xnn_is_hmp_igemm_ukernel(struct xnn_hmp_igemm_ukernel ukernel) { in xnn_is_hmp_igemm_ukernel()
107 struct xnn_hmp_gemm_codegen ukernel = {{ function }}; in xnn_init_hmp_gemm_codegen() local
114 static inline bool xnn_is_hmp_gemm_codegen(struct xnn_hmp_gemm_codegen ukernel) { in xnn_is_hmp_gemm_codegen()
132 struct xnn_hmp_igemm_codegen ukernel = {{ function }}; in xnn_init_hmp_igemm_codegen() local
139 static inline bool xnn_is_hmp_igemm_codegen(struct xnn_hmp_igemm_codegen ukernel) { in xnn_is_hmp_igemm_codegen()
182 xnn_vunary_ukernel_function ukernel; member
244 xnn_spmm_ukernel_function ukernel; member
[all …]
|
| H A D | compute.h | 181 struct xnn_hmp_gemm_ukernel ukernel; member
252 xnn_spmm_ukernel_function ukernel; member
285 struct xnn_hmp_igemm_ukernel ukernel; member
381 struct xnn_hmp_gemm_ukernel ukernel; member
427 struct xnn_hmp_igemm_ukernel ukernel; member
565 xnn_maxpool_ukernel_function ukernel; member
588 xnn_unpool_ukernel_function ukernel; member
747 xnn_gavgpool_cw_ukernel_function ukernel; member
781 xnn_ibilinear_ukernel_function ukernel; member
806 xnn_ibilinear_chw_ukernel_function ukernel; member
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/ |
| H A D | operator-run.c | 36 const pytorch_q8gemm_ukernel_function ukernel; member
88 const pytorch_q8gemm_dq_sparse_ukernel_function ukernel; member
264 const pytorch_q8sum_rows_ukernel_function ukernel; member
309 const pytorch_q8gemm_xzp_ukernel_function ukernel; member
364 const pytorch_q8conv_ukernel_function ukernel; member
540 pytorch_u8maxpool_ukernel_function ukernel; member
715 pytorch_q8vadd_ukernel_function ukernel; member
742 pytorch_q8vadd_ukernel_function ukernel; member
795 pytorch_x8lut_ukernel_function ukernel; member
814 pytorch_x8lut_ukernel_function ukernel; member
[all …]
|
| H A D | conv-run.cc | 27 const pytorch_q8gemm_xzp_ukernel_function ukernel; member
78 const pytorch_q8gemm_ukernel_function ukernel; member
129 const pytorch_q8conv_ukernel_function ukernel; member
179 const pytorch_q8sum_rows_ukernel_function ukernel; member
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuSoftmaxKernel.h | 67 SoftmaxLogits1DMaxKernelPtr ukernel; member
116 SoftmaxLogits1DKernelPtr ukernel; member
|
| H A D | CpuGemmMatrixMultiplyKernel.h | 52 GemmMatrixMulKernelPtr ukernel; member
|
| H A D | CpuAddMulAddKernel.h | 49 AddMulAddKernelPtr ukernel; member
|
| H A D | CpuPool2dKernel.h | 72 PoolingKernelPtr ukernel; member
|
| H A D | CpuDepthwiseConv2dNativeKernel.h | 82 DepthwiseConv2dNativeKernelPtr ukernel; member
|
| H A D | CpuDirectConv2dKernel.h | 77 DirectConv2dKernel_Ptr ukernel; member
|
| H A D | CpuCastKernel.h | 85 CastKernelPtr ukernel; member
|
| H A D | CpuPool3dKernel.h | 78 Pooling3dKernelPtr ukernel; member
|
| H A D | CpuDirectConv3dKernel.h | 83 DirectConv3dKernelPtr ukernel; member
|
| H A D | CpuMaxUnpoolingLayerKernel.h | 78 MaxUnpoolingUKernelPtr ukernel; member
|
| H A D | CpuElementwiseUnaryKernel.h | 74 ElementwiseUnaryUkernelPtr ukernel; member
|
| H A D | CpuGemmMatrixAdditionKernel.h | 54 GemmMatrixAddKernelPtr ukernel; member
|
| H A D | CpuFloorKernel.h | 75 FloorKernelPtr ukernel; member
|
| H A D | CpuActivationKernel.h | 89 ActivationKernelPtr ukernel; member
|
| H A D | CpuAddKernel.h | 47 AddKernelPtr ukernel; member
|
| H A D | CpuScaleKernel.h | 81 ScaleKernelPtr ukernel; member
|
| H A D | CpuElementwiseKernel.h | 62 ElementwiseKernelPtr ukernel; member
|
| /aosp_15_r20/external/XNNPACK/tools/ |
| H A D | generate-transpose-test.py | 269 def generate_test_cases(ukernel, tile_height, tile_width, element_size, isa): argument
295 def generate_memcpy_test_cases(ukernel, tile_height, isa): argument
|
| H A D | generate-raddextexp-test.py | 81 def generate_test_cases(ukernel, elements_tile, isa): argument
|
| /aosp_15_r20/external/XNNPACK/src/operators/ |
| H A D | argmax-pooling-nhwc.c | 34 const struct argmaxpool_parameters* ukernel) in select_ukernel()
221 …const struct argmaxpool_parameters* ukernel = select_ukernel(pooling_size, xnn_params.f32.argmaxpo… in xnn_setup_argmax_pooling2d_nhwc_f32() local
|
| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | argmax-pooling-nhwc.cc | 13 static uint32_t FindMaxSinglePassPoolingSize(const argmaxpool_parameters* ukernel) { in FindMaxSinglePassPoolingSize()
22 static argmaxpool_parameters FindMultiPassMicroKernel(const argmaxpool_parameters* ukernel) { in FindMultiPassMicroKernel()
|