17#ifndef LUCI_INTERPRETER_KERNELS_BATCHMATMUL_H
18#define LUCI_INTERPRETER_KERNELS_BATCHMATMUL_H
28class BatchMatMul :
public KernelWithParams<BatchMatMulParams>
const std::vector< const Tensor * > _inputs
const std::vector< Tensor * > _outputs
const BatchMatMulParams & params() const
void configure() override
void execute() const override
BatchMatMul(const Tensor *x, const Tensor *y, Tensor *output, Tensor *x_tmp, Tensor *y_tmp, const BatchMatMulParams ¶ms)
void BatchMatMul(const BatchMatMulParams ¶ms, const float *lhs_data, const float *rhs_data, float *output_data)