10 #include <arm_compute/runtime/IFunction.h> 11 #include <arm_compute/runtime/Tensor.h> 18 const TensorInfo& inputY,
19 const TensorInfo& output,
20 const BatchMatMulDescriptor& descriptor);
27 virtual void Execute()
const override;
31 std::unique_ptr<arm_compute::IFunction> m_GEMMLayer;
32 std::unique_ptr<arm_compute::IFunction> m_PermuteLayerX;
33 std::unique_ptr<arm_compute::IFunction> m_PermuteLayerY;
37 arm_compute::Tensor m_PermutedTensorX;
38 arm_compute::Tensor m_PermutedTensorY;
arm_compute::Status NeonBatchMatMulValidate(const TensorInfo &inputX, const TensorInfo &inputY, const TensorInfo &output, const BatchMatMulDescriptor &descriptor)
Copyright (c) 2021 ARM Limited and Contributors.
NeonBatchMatMulWorkload(const BatchMatMulQueueDescriptor &descriptor, const WorkloadInfo &info)
virtual void Execute() const override
Contains information about TensorInfos of a layer.