24.02.1
|
Go to the documentation of this file.
24 #ifndef SRC_DYNAMIC_FUSION_SKETCH_GPU_COMPONENTS_CL_CLCOMPONENTDIRECTCONV2D
25 #define SRC_DYNAMIC_FUSION_SKETCH_GPU_COMPONENTS_CL_CLCOMPONENTDIRECTCONV2D
38 namespace experimental
40 namespace dynamic_fusion
45 class Conv2dAttributes;
66 bool _fast_relaxed_math{
true};
71 #ifndef ACL_INTERNAL_TEST_CKW_IN_DF
73 #else // ACL_INTERNAL_TEST_CKW_IN_DF
75 #endif // ACL_INTERNAL_TEST_CKW_IN_DF
142 #ifndef ACL_INTERNAL_TEST_CKW_IN_DF
144 #else // ACL_INTERNAL_TEST_CKW_IN_DF
146 #endif // ACL_INTERNAL_TEST_CKW_IN_DF
154 #ifndef ACL_INTERNAL_TEST_CKW_IN_DF
155 std::unique_ptr<ClTemplateDirectConv2d> _component_writer;
156 #else // ACL_INTERNAL_TEST_CKW_IN_DF
157 std::unique_ptr<GpuCkwDirectConv2d> _component_writer;
158 #endif // ACL_INTERNAL_TEST_CKW_IN_DF
Component specific settings.
GpuComponentType type() const override
Get component type.
GpuComponentType
Component type in the context of fusion Its main purpose is to inform the optimizer how to perform fu...
bool export_to_cl_image() const
Get export_to_cl_image flag.
ClComponentDirectConv2d & operator=(const ClComponentDirectConv2d &component)=delete
Prevent instances of this class from being copied.
An interface used by ClTemplateWriter to write source code for a kernel component.
This is a generic class that packs the arguments of an operator.
ClComponentDirectConv2d(ComponentId id, const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes, const Settings &settings)
Constructor.
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
const IGpuTemplateComponentWriter * template_writer() const override
Get writer for the component.
~ClComponentDirectConv2d() override
Destructor.
virtual const IGpuCkwComponentDriver * ckw_component_driver() const
Properties common to all kernel component types.
bool fast_relaxed_math() const
Get fast_relaxed_math flag.
Properties properties() const
Get properties.
An interface used by GpuCkwDriver to write source code for a kernel component.
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
Copyright (c) 2017-2024 Arm Limited.
An abstract interface of a component.
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
DirectConvComputeKernelInfo direct_conv_descriptor() const
Get direct convolution descriptor.
static Status validate(const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes, const Settings &settings)
Validate the component.
Compute descriptor used by the direct convolution kernel.