21.02
|
#include "src/core/cpu/kernels/CpuPoolingKernel.h"
#include "arm_compute/core/Helpers.h"
#include "arm_compute/core/TensorInfo.h"
#include "arm_compute/core/Validate.h"
#include "arm_compute/core/Window.h"
#include "arm_compute/core/utils/misc/ShapeCalculator.h"
#include "src/core/AccessWindowStatic.h"
#include "src/core/CPP/Validate.h"
#include "src/core/NEON/NEAsymm.h"
#include "src/core/NEON/NEFixedPoint.h"
#include "src/core/NEON/NEMath.h"
#include "src/core/common/Registrars.h"
#include "src/core/cpu/kernels/pooling/neon/list.h"
#include "src/core/helpers/AutoConfiguration.h"
#include "src/core/helpers/WindowHelpers.h"
#include "support/ToolchainSupport.h"
#include "src/core/NEON/wrapper/wrapper.h"
#include <arm_neon.h>
Go to the source code of this file.
Namespaces | |
arm_compute | |
Copyright (c) 2017-2021 Arm Limited. | |
arm_compute::cpu | |
arm_compute::cpu::kernels | |
DataLayout dl |
Definition at line 58 of file CpuPoolingKernel.cpp.
Referenced by arm_compute::error_on_data_layout_not_in(), arm_compute::permute_strides(), and arm_compute::string_from_data_layout().
DataType dt |
Definition at line 57 of file CpuPoolingKernel.cpp.
const PoolingSelectorPtr is_selected |
Definition at line 68 of file CpuPoolingKernel.cpp.
const char* name |
Definition at line 67 of file CpuPoolingKernel.cpp.
Size2D pool_size |
Definition at line 60 of file CpuPoolingKernel.cpp.
Referenced by CpuPoolingKernel::configure(), GCPoolingLayerKernel::configure(), pooling_layer_3(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), and CpuPoolingKernel::run_op().
int pool_stride_x |
Definition at line 59 of file CpuPoolingKernel.cpp.
Referenced by arm_compute::test::validation::reference::pooling_layer_internal(), arm_compute::cpu::poolingMxN_fp32_neon_nhwc(), arm_compute::cpu::poolingMxN_q8_neon_nhwc(), GCPoolingLayerKernel::run(), ClPoolingKernel::run_op(), and CpuPoolingKernel::run_op().
PoolingKernelPtr ukernel |
Definition at line 69 of file CpuPoolingKernel.cpp.