ArmNN
 24.02
BatchMatMul Class Reference

#include <BatchMatMulImpl.hpp>

Public Member Functions

 BatchMatMul (const BatchMatMulDescriptor &params, const TensorInfo &inputXInfo, const TensorInfo &inputYInfo, const TensorInfo &outputInfo, Decoder< float > &inputXDecoder, Decoder< float > &inputYDecoder, Encoder< float > &outputEncoder)
 

Detailed Description

Definition at line 16 of file BatchMatMulImpl.hpp.

Constructor & Destructor Documentation

◆ BatchMatMul()

BatchMatMul ( const BatchMatMulDescriptor params,
const TensorInfo inputXInfo,
const TensorInfo inputYInfo,
const TensorInfo outputInfo,
Decoder< float > &  inputXDecoder,
Decoder< float > &  inputYDecoder,
Encoder< float > &  outputEncoder 
)

Definition at line 15 of file BatchMatMulImpl.cpp.

22  : params(params),
23  inputXInfo(inputXInfo),
24  inputYInfo(inputYInfo),
25  outputInfo(outputInfo),
26  inputXDecoder(inputXDecoder),
27  inputYDecoder(inputYDecoder),
28  outputEncoder(outputEncoder)
29 {
30  inputXData = this->inputXDecoder.DecodeTensor(inputXInfo.GetShape());
31  inputYData = this->inputYDecoder.DecodeTensor(inputYInfo.GetShape());
32  // At this point, we don't touch the input decoders - just the resultant vectors
33 
34  ApplyParams();
35 
36  ApplyBatchMatMul();
37 }

References Decoder< IType >::DecodeTensor(), and TensorInfo::GetShape().


The documentation for this class was generated from the following files:
armnn::Decoder::DecodeTensor
virtual std::vector< float > DecodeTensor(const TensorShape &tensorShape, bool isDepthwise=false)=0
armnn::TensorInfo::GetShape
const TensorShape & GetShape() const
Definition: Tensor.hpp:193