49 …hape shape, PoolingLayerInfo pool_info, DataType data_type, DataLayout data_layout, bool indices =…
54 …_target       = compute_target(shape, pool_info, data_type, data_layout, input_qinfo, output_qinfo…
55 …_reference    = compute_reference(shape, pool_info, data_type, data_layout, input_qinfo, output_qi…
61         const DataLayout data_layout = src.info()->data_layout();  in mix_layout()  local
63 …src.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW);  in mix_layout()
64 …dst.info()->set_data_layout(data_layout == DataLayout::NCHW ? DataLayout::NHWC : DataLayout::NCHW);  in mix_layout()
70         src.info()->set_data_layout(data_layout);  in mix_layout()
71         dst.info()->set_data_layout(data_layout);  in mix_layout()
94                               DataType data_type, DataLayout data_layout,  in compute_target()  argument
99         if(data_layout == DataLayout::NHWC)  in compute_target()
104 …orType        src       = create_tensor<TensorType>(shape, data_type, 1, input_qinfo, data_layout);  in compute_target()
106 …e        dst       = create_tensor<TensorType>(dst_shape, data_type, 1, output_qinfo, data_layout);  in compute_target()
107 …es             = create_tensor<TensorType>(dst_shape, DataType::U32, 1, output_qinfo, data_layout);  in compute_target()
117         add_padding_x({ &src, &dst, &_target_indices }, data_layout);  in compute_target()
143 …ute_reference(TensorShape shape, PoolingLayerInfo info, DataType data_type, DataLayout data_layout,  in compute_reference()  argument
150 …ference::pooling_layer<T>(src, info, output_qinfo, indices ? &_ref_indices : nullptr, data_layout);  in compute_reference()
165 …e, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout)  in setup()  argument
167 … FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_inf…  in setup()
168 …                                                                     data_type, data_layout, true);  in setup()
177 …e, PadStrideInfo pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout)  in setup()  argument
179 … FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_inf…  in setup()
180 …                                                      data_type, data_layout, false, mixed_layout);  in setup()
189 … pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout, bool fp_mixed_…
191 … FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_inf…
192 …                                                                           data_type, data_layout);
201 … pad_stride_info, bool exclude_padding, DataType data_type, DataLayout data_layout = DataLayout::N…
204 … FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, pool_size, data_layout, pad_stride_inf…
205 …                           data_type, data_layout, false, input_qinfo, output_qinfo, mixed_layout);
216 …ype, AccessorType, FunctionType, T>::setup(src_shape, pool_info, data_type, pool_info.data_layout);  in setup()
225 …void setup(TensorShape shape, PoolingType pool_type, DataType data_type, DataLayout data_layout = …
227 …, FunctionType, T>::setup(shape, PoolingLayerInfo(pool_type, data_layout), data_type, data_layout);