Home
last modified time | relevance | path

Searched refs:PoolingType (Results 1 – 25 of 65) sorted by relevance

123

/external/ComputeLibrary/src/core/NEON/kernels/
DNEPoolingLayerKernel.h88 …void pooling2_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type,…
102 …void poolingMxN_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_typ…
110 …void poolingMxN_f32_nhwc(const Window &window_input, const Window &window, PoolingType pooling_typ…
118 …void pooling7_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type,…
126 …void pooling3_f32_nchw(const Window &window_input, const Window &window, PoolingType pooling_type,…
134 …void pooling2_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type,…
155 …void pooling3_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type,…
163 …void poolingMxN_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_typ…
171 …void poolingMxN_f16_nhwc(const Window &window_input, const Window &window, PoolingType pooling_typ…
180 …void pooling2_q8_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, …
[all …]
DNEPoolingLayerKernel.cpp148 PoolingType pool_type = pool_info.pool_type; in validate_arguments()
157 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(pool_type != PoolingType::MAX, "Pooling indices only supported for… in validate_arguments()
160 …ARM_COMPUTE_RETURN_ERROR_ON(pool_type == PoolingType::L2 && is_data_type_quantized(input->data_typ… in validate_arguments()
161 …>data_type()) && !pool_info.exclude_padding && (pool_info.pool_type == PoolingType::AVG) && pool_i… in validate_arguments()
669 …el::pooling2_q8_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, b… in pooling2_q8_nchw()
714 if(pooling_type != PoolingType::MAX) in pooling2_q8_nchw()
796 …l::pooling3_f16_nchw(const Window &window_input, const Window &window, PoolingType pooling_type, b… in pooling3_f16_nchw()
827 if(pooling_type == PoolingType::L2) in pooling3_f16_nchw()
834 if(pooling_type != PoolingType::MAX) in pooling3_f16_nchw()
852 if(pooling_type == PoolingType::L2) in pooling3_f16_nchw()
[all …]
/external/ComputeLibrary/tests/datasets/system_tests/googlenet/inceptionv1/
DGoogLeNetInceptionV1PoolingLayerDataset.h47 …add_config(TensorShape(112U, 112U, 64U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2,… in GoogLeNetInceptionV1PoolingLayerDataset()
49 …add_config(TensorShape(56U, 56U, 192U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in GoogLeNetInceptionV1PoolingLayerDataset()
51 …add_config(TensorShape(28U, 28U, 192U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, … in GoogLeNetInceptionV1PoolingLayerDataset()
53 …add_config(TensorShape(28U, 28U, 256U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, … in GoogLeNetInceptionV1PoolingLayerDataset()
55 …add_config(TensorShape(28U, 28U, 480U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in GoogLeNetInceptionV1PoolingLayerDataset()
57 …add_config(TensorShape(14U, 14U, 480U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, … in GoogLeNetInceptionV1PoolingLayerDataset()
59 …add_config(TensorShape(14U, 14U, 512U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, … in GoogLeNetInceptionV1PoolingLayerDataset()
61 …add_config(TensorShape(14U, 14U, 528U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, … in GoogLeNetInceptionV1PoolingLayerDataset()
63 …add_config(TensorShape(14U, 14U, 832U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in GoogLeNetInceptionV1PoolingLayerDataset()
65 …add_config(TensorShape(7U, 7U, 832U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(1, 1, 1,… in GoogLeNetInceptionV1PoolingLayerDataset()
/external/ComputeLibrary/tests/datasets/system_tests/googlenet/inceptionv4/
DGoogLeNetInceptionV4PoolingLayerDataset.h47 …add_config(TensorShape(147U, 147U, 64U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2,… in GoogLeNetInceptionV4PoolingLayerDataset()
49 …add_config(TensorShape(71U, 71U, 192U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in GoogLeNetInceptionV4PoolingLayerDataset()
51 …add_config(TensorShape(35U, 35U, 384U), PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1, … in GoogLeNetInceptionV4PoolingLayerDataset()
53 …add_config(TensorShape(35U, 35U, 384U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in GoogLeNetInceptionV4PoolingLayerDataset()
55 …add_config(TensorShape(17U, 17U, 1024U), PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1,… in GoogLeNetInceptionV4PoolingLayerDataset()
57 …add_config(TensorShape(17U, 17U, 1024U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2,… in GoogLeNetInceptionV4PoolingLayerDataset()
59 …add_config(TensorShape(8U, 8U, 1536U), PoolingLayerInfo(PoolingType::AVG, 3, PadStrideInfo(1, 1, 1… in GoogLeNetInceptionV4PoolingLayerDataset()
/external/ComputeLibrary/tests/validation/CL/
DPoolingLayer.cpp59 PoolingType::MAX, PoolingType::AVG,
63 combine(combine(framework::dataset::make("PoolingType", { PoolingType::AVG }), framework::dataset::…
68 PoolingType::MAX, PoolingType::AVG,
75 { PoolingType::MAX }),
114 …framework::dataset::make("PoolInfo", { PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, Pa…
115 … PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NCHW, PadStrideInfo(1, 1, 2, 0)),
116 … PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 2)),
117 … PoolingLayerInfo(PoolingType::L2, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
118 … PoolingLayerInfo(PoolingType::AVG, DataLayout::NCHW),
119 … PoolingLayerInfo(PoolingType::MAX, DataLayout::NCHW),
[all …]
/external/ComputeLibrary/tests/validation/GLES_COMPUTE/
DPoolingLayer.cpp76 PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
77 PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
78 PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
79 PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
80 PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NCHW, PadStrideInfo(1, 1, 2, 0)),
81 PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 2)),
82 PoolingLayerInfo(PoolingType::L2, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
83 PoolingLayerInfo(PoolingType::AVG, DataLayout::NCHW),
84 PoolingLayerInfo(PoolingType::MAX, DataLayout::NCHW),
85 PoolingLayerInfo(PoolingType::AVG, DataLayout::NCHW),
/external/ComputeLibrary/tests/datasets/system_tests/vgg/vgg16/
DVGG16PoolingLayerDataset.h46 …add_config(TensorShape(224U, 224U, 64U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2,… in VGG16PoolingLayerDataset()
48 …add_config(TensorShape(112U, 112U, 128U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2… in VGG16PoolingLayerDataset()
50 …add_config(TensorShape(56U, 56U, 256U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, … in VGG16PoolingLayerDataset()
52 …add_config(TensorShape(28U, 28U, 512U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, … in VGG16PoolingLayerDataset()
54 …add_config(TensorShape(14U, 14U, 512U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, … in VGG16PoolingLayerDataset()
/external/ComputeLibrary/tests/datasets/system_tests/yolo/v2/
DYOLOV2PoolingLayerDataset.h46 …add_config(TensorShape(416U, 416U, 32U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2,… in YOLOV2PoolingLayerDataset()
48 …add_config(TensorShape(208U, 208U, 64U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2,… in YOLOV2PoolingLayerDataset()
50 …add_config(TensorShape(104U, 104U, 128U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2… in YOLOV2PoolingLayerDataset()
52 …add_config(TensorShape(52U, 52U, 256U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, … in YOLOV2PoolingLayerDataset()
54 …add_config(TensorShape(26U, 26U, 512U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, … in YOLOV2PoolingLayerDataset()
/external/ComputeLibrary/tests/validation/NEON/
DPoolingLayer.cpp57 …ine(combine(combine(framework::dataset::make("PoolingType", { PoolingType::MAX, PoolingType::AVG }…
95 …framework::dataset::make("PoolInfo", { PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, Pa…
96 … PoolingLayerInfo(PoolingType::AVG, 3, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 0)),
97 … PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NCHW, PadStrideInfo(1, 1, 2, 0)),
98 … PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NCHW, PadStrideInfo(1, 1, 0, 2)),
99 PoolingLayerInfo(PoolingType::AVG, DataLayout::NCHW),
100 PoolingLayerInfo(PoolingType::MAX, DataLayout::NCHW),
101 … PoolingLayerInfo(PoolingType::AVG, 2, DataLayout::NHWC, PadStrideInfo(), false),
102 PoolingLayerInfo(PoolingType::AVG, DataLayout::NCHW),
122 …Small = combine(combine(combine(framework::dataset::make("PoolType", { PoolingType::MAX }), framew…
/external/ComputeLibrary/tests/datasets/
DPoolingTypesDataset.h39 class PoolingTypes final : public framework::dataset::ContainerDataset<std::vector<PoolingType>>
45 PoolingType::MAX, PoolingType::AVG, PoolingType::L2 in PoolingTypes()
DPoolingLayerDataset.h109 …add_config(TensorShape(2U, 3U, 4U, 1U), PoolingLayerInfo(PoolingType::AVG, Size2D(3, 3), DataLayou… in PoolingLayerDatasetSpecial()
110 …add_config(TensorShape(60U, 52U, 3U, 2U), PoolingLayerInfo(PoolingType::AVG, Size2D(100, 100), Dat… in PoolingLayerDatasetSpecial()
112 …add_config(TensorShape(112U, 112U, 32U), PoolingLayerInfo(PoolingType::MAX, 3, DataLayout::NCHW, P… in PoolingLayerDatasetSpecial()
113 …add_config(TensorShape(14U, 14U, 832U), PoolingLayerInfo(PoolingType::MAX, 2, DataLayout::NCHW, Pa… in PoolingLayerDatasetSpecial()
/external/ComputeLibrary/tests/datasets/system_tests/alexnet/
DAlexNetPoolingLayerDataset.h45 …add_config(TensorShape(55U, 55U, 96U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, 0… in AlexNetPoolingLayerDataset()
46 …add_config(TensorShape(27U, 27U, 256U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in AlexNetPoolingLayerDataset()
47 …add_config(TensorShape(13U, 13U, 256U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in AlexNetPoolingLayerDataset()
/external/ComputeLibrary/tests/datasets/system_tests/squeezenet/
DSqueezeNetPoolingLayerDataset.h46 …add_config(TensorShape(111U, 111U, 64U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2,… in SqueezeNetPoolingLayerDataset()
48 …add_config(TensorShape(55U, 55U, 128U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in SqueezeNetPoolingLayerDataset()
50 …add_config(TensorShape(27U, 27U, 256U), PoolingLayerInfo(PoolingType::MAX, 3, PadStrideInfo(2, 2, … in SqueezeNetPoolingLayerDataset()
/external/ComputeLibrary/src/core/CL/kernels/
DCLPoolingLayerKernel.cpp69 …((is_data_type_quantized_asymmetric(input->data_type()) && pool_info.pool_type == PoolingType::L2), in validate_arguments()
76 …ARM_COMPUTE_RETURN_ERROR_ON_MSG(pool_info.pool_type != PoolingType::MAX, "Pooling indices only sup… in validate_arguments()
229 const PoolingType pool_type = pool_info.pool_type; in configure()
259 …if(pool_info.pool_size == Size2D(2, 2) && pool_type == PoolingType::MAX && _indices && is_data_typ… in configure()
296 if(pool_type == PoolingType::MAX) in configure()
324 … const auto use_wider_accumulator = use_fp_mixed_precision && (pool_type != PoolingType::MAX); in configure()
329 if(pool_type != PoolingType::MAX) in configure()
344 …else if(pool_info.pool_size == Size2D(2, 2) && pool_type == PoolingType::MAX && _indices && is_dat… in configure()
368 …on = (data_type == DataType::F16) && pool_info.fp_mixed_precision && pool_type != PoolingType::MAX; in configure()
379 else if(is_data_type_quantized(data_type) && pool_type != PoolingType::MAX) in configure()
[all …]
/external/ComputeLibrary/examples/
Dgraph_vgg16.cpp96 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
113 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
138 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
163 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
188 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
Dgraph_googlenet.cpp89 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, 0, 0, … in do_setup()
106 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, 0, 0, … in do_setup()
109 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, … in do_setup()
115 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, … in do_setup()
118 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, 7, operation_layout, PadStrideInfo(1, 1, … in do_setup()
199 …i_d << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, common_params.data_layout, PadStrideInfo… in get_inception_node()
Dgraph_vgg19.cpp93 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
109 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
139 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
169 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
199 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 2, operation_layout, PadStrideInfo(2, 2, 0, 0))… in do_setup()
Dgraph_squeezenet_v1_1.cpp86 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, 0, 0, … in do_setup()
103 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, … in do_setup()
120 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, … in do_setup()
160 … << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, operation_layout)).set_name("pool10") in do_setup()
Dgraph_squeezenet.cpp86 …<< PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, 0, 0, … in do_setup()
111 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, … in do_setup()
144 …graph << PoolingLayer(PoolingLayerInfo(PoolingType::MAX, 3, operation_layout, PadStrideInfo(2, 2, … in do_setup()
160 … << PoolingLayer(PoolingLayerInfo(PoolingType::AVG, operation_layout)).set_name("pool10") in do_setup()
/external/armnn/src/backends/aclCommon/
DArmComputeUtils.hpp119 inline arm_compute::PoolingType ConvertPoolingAlgorithmToAclPoolingType(PoolingAlgorithm poolingAlg… in ConvertPoolingAlgorithmToAclPoolingType()
121 using arm_compute::PoolingType; in ConvertPoolingAlgorithmToAclPoolingType()
125 case PoolingAlgorithm::Max: return PoolingType::MAX; in ConvertPoolingAlgorithmToAclPoolingType()
126 case PoolingAlgorithm::Average: return PoolingType::AVG; in ConvertPoolingAlgorithmToAclPoolingType()
127 case PoolingAlgorithm::L2: return PoolingType::L2; in ConvertPoolingAlgorithmToAclPoolingType()
/external/ComputeLibrary/tests/datasets/system_tests/lenet5/
DLeNet5PoolingLayerDataset.h45 …add_config(TensorShape(24U, 24U, 20U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, 0… in LeNet5PoolingLayerDataset()
46 …add_config(TensorShape(8U, 8U, 50U), PoolingLayerInfo(PoolingType::MAX, 2, PadStrideInfo(2, 2, 0, … in LeNet5PoolingLayerDataset()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpooling_test_util.cc38 attr.type = PoolingType::AVERAGE; in AveragePoolingTest()
69 attr.type = PoolingType::AVERAGE; in AveragePoolingNonEmptyPaddingTest()
101 attr.type = PoolingType::MAX; in MaxPoolingTest()
132 attr.type = PoolingType::MAX; in MaxPoolingIndicesTest()
/external/ComputeLibrary/tests/validation/fixtures/
DPoolingLayerFixture.h142 …void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_in… in setup()
154 …void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_in… in setup()
166 …void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_in…
178 …void setup(TensorShape shape, PoolingType pool_type, Size2D pool_size, PadStrideInfo pad_stride_in…
201 …void setup(TensorShape shape, PoolingType pool_type, DataType data_type, DataLayout data_layout = …
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dpooling_test.cc58 attr.type = PoolingType::MAX; in TEST()
88 attr.type = PoolingType::MAX; in TEST()
114 attr.type = PoolingType::AVERAGE; in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d.h25 enum PoolingType { MAX, AVG }; enum
27 template <typename Device, typename T, PoolingType Type>

123