24.04
|
Go to the documentation of this file.
34 namespace experimental
36 namespace dynamic_fusion
53 "input and target data types should be different");
56 if (
dst->total_size() > 0)
60 "dst and target data types should be same");
71 _component_writer{std::make_unique<GpuCkwCast>(
id, tensors, attributes)}
82 return _component_writer.get();
SimpleTensor< float > src
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(...)
This is a generic class that packs the arguments of an operator.
~ClComponentCast() override
Destructor.
Properties common to all kernel component types.
#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.
An interface used by GpuCkwDriver to write source code for a kernel component.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
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.
An abstract interface of a component.
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
const IGpuCkwComponentDriver * ckw_component_driver() const override
Get writer for the component.
Store the tensor's metadata.
ClComponentCast(ComponentId id, const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes, const Settings &settings)
Constructor.
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.
CastAttributes & data_type(const DataType &data_type)
Set Data Type to be casted to.
Component specific settings.