Searched refs:GetDepthIndex (Results 1 – 6 of 6) sorted by relevance
26 unsigned int GetDepthIndex() const { return m_DepthIndex; } in GetDepthIndex() function in armnnUtils::DataLayoutIndexed
51 unsigned int inDepth = inputShape[dimensionIndices.GetDepthIndex()]; in InferOutputShapes()
78 unsigned int inDepth = inputShape[dataLayoutIndex.GetDepthIndex()]; in InferOutputShapes()
39 const unsigned int depthIndex = dataLayoutIndexed.GetDepthIndex(); in Convolve3d()
182 auto depthIndex = dataLayout.GetDepthIndex(); in Pooling3d()
59 auto depthIndex = dimensionIndices.GetDepthIndex(); in SimplePooling3dTestImpl()