24.02.1
|
Go to the documentation of this file.
32 namespace experimental
34 namespace dynamic_fusion
static ITensorInfo * create_op(GpuWorkloadSketch &sketch, ITensorInfo *lhs, ITensorInfo *rhs, const ElementwiseBinaryCommonAttributes &attributes)
Create an operator and fuse it into the workload sketch.
ElementwiseBinaryCommonAttributes & operation(const ElementwiseBinaryCommonAttributes::ElementwiseOp &operation)
Set operation.
#define ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(t, c,...)
static Status validate_op(const GpuWorkloadSketch &sketch, const ITensorInfo *rhs, const ITensorInfo *lhs)
Validate the operator and check if the its configuration is supported and if it can be fused into the...
A descriptor of a workload of operators.
@ U8
unsigned 8-bit number
@ S16
signed 16-bit number
Provide context necessary for the creation and configuration of a workload e.g.
static Status is_supported_op(const GpuWorkloadContext &context, const ITensorInfo *lhs, const ITensorInfo *rhs)
Check if the operator configuration is supported, irrespective of fusion.
virtual DataType data_type() const =0
Data type used for each element of the tensor.
static ITensorInfo * create_op(GpuWorkloadSketch &sketch, ITensorInfo *lhs, ITensorInfo *rhs)
Create an operator and fuse it into the workload sketch.
#define ARM_COMPUTE_RETURN_ERROR_ON_MSG(cond, msg)
If the condition is true, an error is returned.
Copyright (c) 2017-2024 Arm Limited.
@ F16
16-bit floating-point number
@ S32
signed 32-bit number
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
Store the tensor's metadata.
@ F32
32-bit floating-point number
static Status validate_op(const GpuWorkloadSketch &sketch, const ITensorInfo *rhs, const ITensorInfo *lhs, const ElementwiseBinaryCommonAttributes &attributes)
Validate the operator and check if it can be fused into the workload sketch.
static Status is_supported_op(const GpuWorkloadContext &context, const ITensorInfo *lhs, const ITensorInfo *rhs, const ElementwiseBinaryCommonAttributes &attributes)
Check if the operator configuration is supported, irrespective of fusion.