33 inputXInfo(inputXInfo),
34 inputYInfo(inputYInfo),
35 outputInfo(outputInfo),
36 inputXDecoder(inputXDecoder),
37 inputYDecoder(inputYDecoder),
38 outputEncoder(outputEncoder)
43 void RecurseBMM(std::vector<unsigned int>& curIdx,
unsigned int curDim);
47 std::pair<std::pair<unsigned int, unsigned int>, std::pair<unsigned int, unsigned int>>& axesToMul);
70 std::vector<float> inputXData;
71 std::vector<float> inputYData;
unsigned int CalcFlatIdx(DataSlot type, const std::vector< unsigned int > &idx)
void AdjustToSafeIdx(DataSlot type, std::vector< unsigned int > &idx)
Copyright (c) 2021 ARM Limited and Contributors.
std::string StringifyVec(const std::vector< T > &vec)
BatchMatMul(const BatchMatMulDescriptor ¶ms, const TensorInfo &inputXInfo, const TensorInfo &inputYInfo, const TensorInfo &outputInfo, Decoder< float > &inputXDecoder, Decoder< float > &inputYDecoder, Encoder< float > &outputEncoder)
A BatchMatMulDescriptor for the BatchMatMul operator.
void RecurseBMM(std::vector< unsigned int > &curIdx, unsigned int curDim)
void AdjustAxesToMulForUnequalRanks(std::pair< std::pair< unsigned int, unsigned int >, std::pair< unsigned int, unsigned int >> &axesToMul)
void SetValueAt(float value, DataSlot type, std::vector< unsigned int > idx)
float GetValueAt(DataSlot type, std::vector< unsigned int > idx)