Home
last modified time | relevance | path

Searched refs:dimsToSkip (Results 1 – 2 of 2) sorted by relevance

/external/armnn/src/backends/aclCommon/
DArmComputeTensorUtils.cpp153 unsigned int dimsToSkip = 0; in ReduceDimsForACL() local
157 dimsToSkip = tensorShape.GetNumDimensions() - dimensions; in ReduceDimsForACL()
164 if (tensorShape[i] == 1 && dimsSkipped < dimsToSkip && !insertRemainder) in ReduceDimsForACL()
/external/armnn/src/armnnUtils/
DTensorUtils.cpp114 unsigned int dimsToSkip = tensorShape.GetNumDimensions() - dimensions; in ReduceDims() local
120 if (tensorShape[i] == 1 && dimsSkipped < dimsToSkip && !insertRemainder) in ReduceDims()