ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALBatchMatMul.h File Reference
#include "PALUtils.h"
#include "core/OMRuntimeShape.h"

Go to the source code of this file.

Namespaces

namespace  onert_micro
 
namespace  onert_micro::execute
 
namespace  onert_micro::execute::pal
 

Functions

template<typename T , typename AccumT >
OMStatus onert_micro::execute::pal::BatchMatMul (const OMRuntimeShape &lhs_shape, const T *lhs_data, const OMRuntimeShape &rhs_shape, const T *rhs_data, const OMRuntimeShape &output_shape, T *output_data)