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;