24.04
|
Go to the documentation of this file.
39 namespace experimental
41 namespace dynamic_fusion
54 "Unsupported Pooling type");
60 "Pooling region that is entirely outside input tensor is unsupported");
95 _component_writer{std::make_unique<GpuCkwPool2d>(
id, tensors, attributes, settings)}
103 return _component_writer.get();
Operator backend specific settings.
SimpleTensor< float > src
@ NHWC
Num samples, height, width, channels.
TensorShape compute_pool_shape(const ITensorInfo &input, PoolingLayerInfo pool_info)
Calculate the output pool shape of a tensor.
This is a generic class that packs the arguments of an operator.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(...)
#define ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(t, c,...)
PoolingLayerInfo convert_pool_attr_to_pool_info(const Pool2dAttributes &pool_attr, bool mixed_precision=false, DataLayout data_layout=DataLayout::NHWC)
Inline function to convert Pool2dAttributes to PoolingLayerInfo.
ClComponentPool2d(ComponentId id, const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes, const Settings &settings)
Constructor.
PoolingType pool_type() const
Properties common to all kernel component types.
#define ARM_COMPUTE_ERROR_ON_MSG(cond, msg)
#define ARM_COMPUTE_RETURN_ERROR_ON(cond)
If the condition is true, an error is returned.
#define ARM_COMPUTE_RETURN_ERROR_ON_F16_UNSUPPORTED(tensor)
Properties properties() const
Get properties.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_LAYOUT(...)
An interface used by GpuCkwDriver to write source code for a kernel component.
const IGpuCkwComponentDriver * ckw_component_driver() const override
Get GPU kernel writer for the component.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
#define ARM_COMPUTE_RETURN_ERROR_ON_DATA_LAYOUT_NOT_IN(t,...)
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
#define ARM_COMPUTE_RETURN_ERROR_ON_MSG(cond, msg)
If the condition is true, an error is returned.
Copyright (c) 2017-2024 Arm Limited.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(...)
@ F16
16-bit floating-point number
bool is_pool_region_entirely_outside_input(const PoolingLayerInfo &info)
Check if the pool region is entirely outside the input tensor.
An abstract interface of a component.
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
@ F32
32-bit floating-point number
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
static Status validate(const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes, const Settings &settings)
Validate the component.
ComponentId id() const
Get component id.
~ClComponentPool2d() override
Destructor.