Searched refs:GetNumElementsAfter (Results 1 – 3 of 3) sorted by relevance
234 unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int axis) in GetNumElementsAfter() function256 unsigned int axisFactor = GetNumElementsAfter(info.GetShape(), quantizationDim.value()) ; in GetPerAxisParams()283 auto axisFactor = armnnUtils::GetNumElementsAfter(tensorInfo.GetShape(), axis); in ToFloatArray()
61 unsigned int GetNumElementsAfter(const armnn::TensorShape& shape, unsigned int axis);
717 m_AxisFactor = armnnUtils::GetNumElementsAfter(tensorShape, axis); in PerAxisIterator()