24.04
|
#include <GpuMatMul.h>
Public Types | |
using | Attributes = MatMulAttributes |
using | Settings = GpuMatMulSettings |
Static Public Member Functions | |
static ITensorInfo * | create_op (GpuWorkloadSketch &sketch, ITensorInfo *lhs, ITensorInfo *rhs, const Attributes &attributes, const Settings &settings) |
static Status | is_supported_op (const GpuWorkloadContext &context, const ITensorInfo *lhs, const ITensorInfo *rhs, const Attributes &attributes, const Settings &settings) |
static Status | validate_op (const GpuWorkloadSketch &sketch, const ITensorInfo *lhs, const ITensorInfo *rhs, const Attributes &attributes, const Settings &settings) |
Operator interface.
Definition at line 68 of file GpuMatMul.h.
using Attributes = MatMulAttributes |
Definition at line 72 of file GpuMatMul.h.
using Settings = GpuMatMulSettings |
Definition at line 74 of file GpuMatMul.h.
|
static |
Definition at line 191 of file GpuMatMul.cpp.
References arm_compute::ACL_DST_0, arm_compute::ACL_SRC_0, arm_compute::ACL_SRC_1, ArgumentPack< T >::add_const_tensor(), GpuOperatorGroup::add_operator(), ARM_COMPUTE_ERROR, ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, ARM_COMPUTE_LOG_PARAMS, GpuWorkloadSketch::Implementation::component_graph(), GpuWorkloadSketch::Implementation::context(), GpuWorkloadSketch::Implementation::create_virtual_tensor(), 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, and GpuMatMul::validate_op().
|
static |
Definition at line 151 of file GpuMatMul.cpp.
References arm_compute::test::validation::context.
|
static |
Definition at line 160 of file GpuMatMul.cpp.
References arm_compute::ACL_DST_0, arm_compute::ACL_SRC_0, arm_compute::ACL_SRC_1, ArgumentPack< T >::add_const_tensor(), ARM_COMPUTE_RETURN_ERROR_ON, ARM_COMPUTE_RETURN_ERROR_ON_MSG, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, GpuWorkloadSketch::gpu_context(), ITensorInfo::has_valid_id(), GpuWorkloadSketch::implementation(), GpuOperatorGroup::new_operator(), GpuWorkloadSketch::Implementation::operator_group(), arm_compute::test::validation::sketch, and GpuOperatorGroup::try_add_operator().
Referenced by GpuMatMul::create_op().