24.04
|
#include <GpuMul.h>
Static Public Member Functions | |
static ITensorInfo * | create_op (GpuWorkloadSketch &sketch, ITensorInfo *lhs, ITensorInfo *rhs) |
Create an operator and fuse it into the workload sketch. More... | |
static Status | is_supported_op (const GpuWorkloadContext &context, const ITensorInfo *lhs, const ITensorInfo *rhs) |
Check if the operator configuration is supported, irrespective of fusion. More... | |
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 workload sketch. More... | |
|
static |
Create an operator and fuse it into the workload sketch.
sketch
remains unchanged and valid.Valid data type configurations:
lhs | rhs | dst |
---|---|---|
F16 | F16 | F16 |
F32 | F32 | F32 |
Valid data layouts:
[in,out] | sketch | Workload sketch into which the operator will be fused |
[in] | lhs | Left hand side tensor info. Data types supported: F16/F32. |
[in] | rhs | Right hand side tensor info. Data types supported: Same as lhs . |
Definition at line 60 of file GpuMul.cpp.
References GpuElementwiseBinaryCommon::create_op(), ElementwiseBinaryCommonAttributes::Mul, ElementwiseBinaryCommonAttributes::operation(), and arm_compute::test::validation::sketch.
Referenced by arm_compute::test::validation::TEST_CASE().
|
static |
Check if the operator configuration is supported, irrespective of fusion.
[in] | context | Workload context within which the operator is running |
[in] | lhs | Left hand side tensor info. Data types supported: F16/F32. |
[in] | rhs | Right hand side tensor info. Data types supported: Same as lhs . |
Definition at line 48 of file GpuMul.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN, ARM_COMPUTE_RETURN_ERROR_ON_MSG, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, arm_compute::test::validation::context, ITensorInfo::data_type(), arm_compute::F16, arm_compute::F32, GpuElementwiseBinaryCommon::is_supported_op(), ElementwiseBinaryCommonAttributes::Mul, and ElementwiseBinaryCommonAttributes::operation().
|
static |
Validate the operator and check if the configuration is supported and if it can be fused into the workload sketch.
Parameters are similar to GpuMul::create_op()
Definition at line 36 of file GpuMul.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN, ARM_COMPUTE_RETURN_ERROR_ON_MSG, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, ITensorInfo::data_type(), arm_compute::F16, arm_compute::F32, ElementwiseBinaryCommonAttributes::Mul, ElementwiseBinaryCommonAttributes::operation(), arm_compute::test::validation::sketch, and GpuElementwiseBinaryCommon::validate_op().
Referenced by arm_compute::test::validation::TEST_CASE().