24.04
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWPOOL2D_H
25 #define ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWPOOL2D_H
35 namespace experimental
37 namespace dynamic_fusion
78 #endif // ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWPOOL2D_H
Operator backend specific settings.
std::string get_name(const ComponentGroup &comp_group) const override
Generate the name of the component.
Window get_window() const override
Generate the execution window for the component.
Pool2dAttributes Attributes
Attributes are a set of backend-agnostic parameters that define what a component does.
This is a generic class that packs the arguments of an operator.
~GpuCkwPool2d() override=default
Destructor.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(GpuCkwPool2d)
A table of all the variables used in the kernel.
An interface used by GpuCkwDriver to write source code for a kernel component.
GpuPool2dSettings Settings
Settings are a set of backend-specific parameters that influence the implementation of a component.
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
GpuCkwPool2d(ComponentId id, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes, const Settings &settings)
Constructor.
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
virtual void write_component_code(const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, GpuCkwScopedKernelWriter writer) const override
Generate kernel component code.
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
A group of gpu kernel components to be fused together PRECONDITIONS:
Store the tensor's metadata.
Helper to automatically manage kernel writer ID space.