Home
last modified time | relevance | path

Searched refs:compute_pool_shape (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/
H A DGpuPool2d.cpp75 …auto shape = misc::shape_calculator::compute_pool_shape(*src, convert_pool_attr_to_pool_info(attri… in validate_op()
110 …auto shape = misc::shape_calculator::compute_pool_shape(*src, convert_pool_attr_to_pool_info(attri… in is_supported_op()
149 …auto shape = misc::shape_calculator::compute_pool_shape(*src, convert_pool_attr_to_pool_info(attri… in create_op()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClPool2dKernel.cpp76 TensorInfo idx_info(TensorInfo(compute_pool_shape(*src, pool_info), 1, DataType::U32)); in validate_arguments()
86 TensorInfo out_info(TensorInfo(compute_pool_shape(*src, pool_info), 1, dst->data_type())); in validate_arguments()
107 TensorShape out_shape = compute_pool_shape(*src, pool_info); in configure()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuPool2dKernel.cpp171 TensorInfo out_info(TensorInfo(compute_pool_shape(*src, pool_info), 1, dst->data_type())); in validate_arguments()
210 auto_init_if_empty(*dst, src->clone()->set_tensor_shape(compute_pool_shape(*src, pool_info))); in validate_and_configure_window()
214 auto_init_if_empty(*indices, (src->clone()->set_tensor_shape(compute_pool_shape(*src, in validate_and_configure_window()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DPoolingLayer.cpp44 …SimpleTensor<T> dst{ compute_pool_shape(TensorInfo(src.shape(), 1, src.data_type()), info), src.da… in pooling_layer_internal()
45 …auto pooled_shape = compute_pool_shape(TensorInfo(src.shape(), 1, src.data_type()), inf… in pooling_layer_internal()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/
H A DClComponentPool2d.cpp73 …misc::shape_calculator::compute_pool_shape(*src, convert_pool_attr_to_pool_info(attributes, settin… in validate()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DMaxUnpoolingLayerFixture.h96 …const TensorShape dst_shape = misc::shape_calculator::compute_pool_shape(*(src.info()), pool_info); in compute_target()
H A DPoolingLayerFixture.h105 … const TensorShape dst_shape = misc::shape_calculator::compute_pool_shape(*(src.info()), info); in compute_target()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/
H A DCpuPool2dAssemblyWrapperKernel.cpp50 auto_init_if_empty(*dst, src->clone()->set_tensor_shape(compute_pool_shape(*src, info))); in configure()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h801 inline TensorShape compute_pool_shape(const ITensorInfo &input, PoolingLayerInfo pool_info) in compute_pool_shape() function