10 #include <arm_compute/runtime/IFunction.h>
11 #include <arm_compute/runtime/CL/CLTensor.h>
17 const TensorInfo& inputY,
18 const TensorInfo& output,
19 const BatchMatMulDescriptor& descriptor);
26 const arm_compute::CLCompileContext& clCompileContext);
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::CLTensor m_PermutedTensorX;
38 arm_compute::CLTensor m_PermutedTensorY;