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 D | GpuPool2d.cpp | 75 …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 D | ClPool2dKernel.cpp | 76 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 D | CpuPool2dKernel.cpp | 171 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 D | PoolingLayer.cpp | 44 …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 D | ClComponentPool2d.cpp | 73 …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 D | MaxUnpoolingLayerFixture.h | 96 …const TensorShape dst_shape = misc::shape_calculator::compute_pool_shape(*(src.info()), pool_info); in compute_target()
|
H A D | PoolingLayerFixture.h | 105 … 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 D | CpuPool2dAssemblyWrapperKernel.cpp | 50 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 D | ShapeCalculator.h | 801 inline TensorShape compute_pool_shape(const ITensorInfo &input, PoolingLayerInfo pool_info) in compute_pool_shape() function
|