Searched refs:create_arm_pooling (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/ |
| H A D | CpuPool2dAssemblyWrapperKernel.cpp | 64 create_arm_pooling<uint8_t, uint8_t>(src, dst, info, cpu_info); in configure()
74 create_arm_pooling<int8_t, int8_t>(src, dst, info, cpu_info); in configure()
79 create_arm_pooling<float16_t, float16_t>(src, dst, info, cpu_info); in configure()
83 create_arm_pooling<float, float>(src, dst, info, cpu_info); in configure()
189 void CpuPool2dAssemblyWrapperKernel::create_arm_pooling(const ITensorInfo *src, ITensorInfo *dst, c… in create_arm_pooling() function in arm_compute::cpu::kernels::CpuPool2dAssemblyWrapperKernel
|
| H A D | CpuPool2dAssemblyWrapperKernel.h | 104 …void create_arm_pooling(const ITensorInfo *src, ITensorInfo *dst, const PoolingLayerInfo &info, co…
|