Searched refs:GetNumElementsBetween (Results 1 – 5 of 5) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | ArgMinMax.cpp | 23 …const unsigned int outerElements = armnnUtils::GetNumElementsBetween(inputTensorInfo.GetShape(), 0… in ArgMinMax()
25 const unsigned int innerElements = armnnUtils::GetNumElementsBetween(inputTensorInfo.GetShape(), in ArgMinMax()
|
| D | Softmax.cpp | 29 const unsigned int outerSize = armnnUtils::GetNumElementsBetween(inputShape, 0, uAxis); in Softmax()
31 const unsigned int innerSize = armnnUtils::GetNumElementsBetween(inputShape, in Softmax()
|
| D | LogSoftmax.cpp | 46 const unsigned int outerSize = armnnUtils::GetNumElementsBetween(inputShape, 0, uAxis); in LogSoftmax()
48 const unsigned int innerSize = armnnUtils::GetNumElementsBetween(inputShape, in LogSoftmax()
|
| /external/armnn/include/armnnUtils/ |
| D | TensorUtils.hpp | 55 unsigned int GetNumElementsBetween(const armnn::TensorShape& shape,
|
| /external/armnn/src/armnnUtils/ |
| D | TensorUtils.cpp | 207 unsigned int GetNumElementsBetween(const TensorShape& shape, in GetNumElementsBetween() function
|