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.

    : params(params),
      inputXInfo(inputXInfo),
      inputYInfo(inputYInfo),
      outputInfo(outputInfo),
      inputXDecoder(inputXDecoder),
      inputYDecoder(inputYDecoder),
      outputEncoder(outputEncoder)
{
    inputXData = this->inputXDecoder.DecodeTensor(inputXInfo.GetShape());
    inputYData = this->inputYDecoder.DecodeTensor(inputYInfo.GetShape());
    // At this point, we don't touch the input decoders - just the resultant vectors
 
    ApplyParams();
 
    ApplyBatchMatMul();
}

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

---

The documentation for this class was generated from the following files:

• src/backends/reference/workloads/BatchMatMulImpl.hpp
• src/backends/reference/workloads/BatchMatMulImpl.cpp