24.04
|
#include <GpuOutput.h>
Static Public Member Functions | |
static void | create_op (GpuWorkloadSketch &sketch, ITensorInfo *src, ITensorInfo *dst) |
Create an operator and fuse it into the workload sketch. More... | |
static Status | is_supported_op (const GpuWorkloadContext &context, const ITensorInfo *src, const ITensorInfo *dst) |
Check if the operator configuration is supported, irrespective of fusion. More... | |
static Status | validate_op (const GpuWorkloadSketch &sketch, const ITensorInfo *src, const ITensorInfo *dst) |
Validate the operator and check if the its configuration is supported and if it can be fused into the workload sketch. More... | |
Operator interface.
Definition at line 41 of file GpuOutput.h.
|
static |
Create an operator and fuse it into the workload sketch.
sketch
remains unchanged and valid.Valid data type configurations:
Valid data layouts:
[in,out] | sketch | Workload sketch into which the operator will be fused. |
[in,out] | src | Source tensor info. |
[in,out] | dst | Destination tensor info. If an uninitialized ITensorInfo is passed in, it will be auto-initialized. |
Definition at line 88 of file GpuOutput.cpp.
References arm_compute::ACL_DST_0, arm_compute::ACL_SRC_0, ArgumentPack< T >::add_const_tensor(), GpuOperatorGroup::add_operator(), ARM_COMPUTE_ERROR, ARM_COMPUTE_ERROR_ON, ARM_COMPUTE_ERROR_THROW_ON, ARM_COMPUTE_LOG_PARAMS, arm_compute::auto_init_if_empty(), GpuWorkloadSketch::Implementation::component_graph(), GpuWorkloadSketch::Implementation::context(), arm_compute::test::validation::dst, GpuWorkloadSketch::implementation(), GpuOperatorGroup::new_operator(), arm_compute::experimental::dynamic_fusion::OpenCL, GpuWorkloadSketch::Implementation::operator_group(), UnitWorkloadStage::Run, arm_compute::test::validation::sketch, arm_compute::test::validation::src, KernelProperties::stage(), and GpuOutput::validate_op().
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] | src | Source tensor info. |
[in] | dst | Destination tensor info. |
Definition at line 45 of file GpuOutput.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES, ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, ARM_COMPUTE_UNUSED, arm_compute::auto_init_if_empty(), arm_compute::test::validation::context, arm_compute::test::validation::dst, and arm_compute::test::validation::src.
Referenced by GpuOutput::validate_op().
|
static |
Validate the operator and check if the its configuration is supported and if it can be fused into the workload sketch.
Parameters are similar to GpuOutput::create_op().
Definition at line 60 of file GpuOutput.cpp.
References arm_compute::ACL_DST_0, arm_compute::ACL_SRC_0, ArgumentPack< T >::add_const_tensor(), ARM_COMPUTE_RETURN_ERROR_ON, ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES, ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES, ARM_COMPUTE_RETURN_ERROR_ON_MSG, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, arm_compute::auto_init_if_empty(), arm_compute::test::validation::dst, GpuWorkloadSketch::gpu_context(), GpuWorkloadSketch::implementation(), arm_compute::experimental::dynamic_fusion::is_alloc_tensor(), GpuOutput::is_supported_op(), GpuOperatorGroup::new_operator(), GpuWorkloadSketch::Implementation::operator_group(), arm_compute::test::validation::sketch, and arm_compute::test::validation::src.
Referenced by GpuOutput::create_op(), and arm_compute::test::validation::TEST_CASE().