24 #ifndef ARM_COMPUTE_TEST_LENET5_POOLING_LAYER_DATASET 25 #define ARM_COMPUTE_TEST_LENET5_POOLING_LAYER_DATASET 45 add_config(
TensorShape(24
U, 24
U, 20
U),
TensorShape(12
U, 12
U, 20
U),
PoolingLayerInfo(
PoolingType::MAX, 2,
PadStrideInfo(2, 2, 0, 0)));
46 add_config(
TensorShape(8
U, 8
U, 50
U),
TensorShape(4
U, 4
U, 50
U),
PoolingLayerInfo(
PoolingType::MAX, 2,
PadStrideInfo(2, 2, 0, 0)));
LeNet5PoolingLayerDataset()
This file contains all available output stages for GEMMLowp on OpenCL.
Padding and stride information class.
void add_config(TensorShape src, TensorShape dst, PoolingLayerInfo info)
Pooling Layer Information class.