Searched refs:SetQuantizationDim (Results 1 – 10 of 10) sorted by relevance
85 infoB.SetQuantizationDim(Optional<unsigned int>(1));111 infoA.SetQuantizationDim(Optional<unsigned int>(1));118 infoB.SetQuantizationDim(Optional<unsigned int>(2));247 tensorInfo0.SetQuantizationDim(Optional<unsigned int>(1));
381 SetQuantizationDim(MakeOptional<unsigned int>(quantizationDim)); in TensorInfo()395 SetQuantizationDim(MakeOptional<unsigned int>(quantizationDim)); in TensorInfo()499 void TensorInfo::SetQuantizationDim(const Optional<unsigned int>& quantizationDim) in SetQuantizationDim() function in armnn::TensorInfo
125 outInfo.SetQuantizationDim(mappings[info.GetQuantizationDim().value()]); in Permuted()
215 void SetQuantizationDim(const Optional<unsigned int>& quantizationDim);
193 ret.SetQuantizationDim(MakeOptional<unsigned int>(perAxisQuantParams.channelDim)); in GetTensorInfoForOperand()282 ret.SetQuantizationDim(MakeOptional<unsigned int>(perAxisQuantParams.channelDim)); in GetTensorInfoForOperand()
498 biasInfo.SetQuantizationDim(0); in SanitizeBiasQuantizationScale()
430 …ret.SetQuantizationDim(armnn::numeric_cast<unsigned int>(affineQuantization->quantized_dimension)); in GetTensorInfoForTfLiteTensor()
510 …ret.SetQuantizationDim(armnn::numeric_cast<unsigned int>(affineQuantization->quantized_dimension)); in GetTensorInfoForTfLiteOpaqueTensor()
179 ret.SetQuantizationDim(MakeOptional<unsigned int>(perAxisQuantParams.channelDim)); in GetTensorInfoForOperand()
437 biasInfo.SetQuantizationDim(0); in SanitizeBiasQuantizationScale()