24.04
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWACTIVATION
25 #define ACL_SRC_DYNAMIC_FUSION_SKETCH_GPU_CKW_DRIVER_COMPONENTS_GPUCKWACTIVATION
33 namespace experimental
35 namespace dynamic_fusion
GpuCkwActivation(ComponentId id, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes)
Constructor.
ActivationLayerInfo 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.
Activation Layer Information class.
A table of all the variables used in the kernel.
virtual void write_component_code(const ComponentGroup &comp_group, GpuCkwVariableTable &vtable, GpuCkwScopedKernelWriter writer) const override
Generate kernel component code.
An interface used by GpuCkwDriver to write source code for a kernel component.
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
Describe a multidimensional execution window.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(GpuCkwActivation)
Copyright (c) 2017-2024 Arm Limited.
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
A group of gpu kernel components to be fused together PRECONDITIONS:
Store the tensor's metadata.
Window get_window() const override
Generate the execution window for the component.
~GpuCkwActivation() override=default
Destructor.
Helper to automatically manage kernel writer ID space.