24.04
|
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,...)
A descriptor of a workload of operators.
Provide context necessary for the creation and configuration of a workload e.g.
static ITensorInfo * create_op(GpuWorkloadSketch &sketch, ITensorInfo *lhs, ITensorInfo *rhs)
Create an operator and fuse it into the workload sketch.
virtual DataType data_type() const =0
Data type used for each element of the tensor.
static Status is_supported_op(const GpuWorkloadContext &context, const ITensorInfo *lhs, const ITensorInfo *rhs)
Check if the operator configuration is supported, irrespective of fusion.
#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
#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)
Validate the operator and check if the configuration is supported and if it can be fused into the wor...
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.