24.02.1
|
Go to the documentation of this file.
37 namespace experimental
39 namespace dynamic_fusion
81 _component_writer{std::make_unique<ClTemplateLogits1DNorm>(
id, tensors, attributes)}
91 return _component_writer.get();
SimpleTensor< float > src
virtual const TensorShape & tensor_shape() const =0
Size for each dimension of the tensor.
Attributes are backend-agnostic parameters (in addition to the input/output tensors) of an operator.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(...)
~ClComponentLogits1DNorm() override
Destructor.
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.
#define ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(...)
#define ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(t, c,...)
Properties common to all kernel component types.
#define ARM_COMPUTE_RETURN_ERROR_ON(cond)
If the condition is true, an error is returned.
size_t total_size() const
Collapses all dimensions to a single linear total size.
#define ARM_COMPUTE_RETURN_ERROR_ON_F16_UNSUPPORTED(tensor)
Properties properties() const
Get properties.
ClComponentLogits1DNorm(ComponentId id, const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes)
Constructor.
#define ARM_COMPUTE_UNUSED(...)
To avoid unused variables warnings.
SoftmaxAttributes & is_log_softmax(bool is_log)
Set is_log.
int32_t ComponentId
Uniquely identifies a kernel component within a workload.
Copyright (c) 2017-2024 Arm Limited.
@ F16
16-bit floating-point number
static Status validate(const Properties &properties, const ArgumentPack< ITensorInfo > &tensors, const Attributes &attributes)
Validate the component.
const IGpuTemplateComponentWriter * template_writer() const override
Get template writer for the component.
bool is_data_type_float(DataType dt)
Check if a given data type is of floating point type.
An abstract interface of a component.
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
Store the tensor's metadata.
@ F32
32-bit floating-point number
ArgumentPack< ITensorInfo > tensors() const
Get tensor arguments.
ComponentId id() const
Get component id.