Home
last modified time | relevance | path

Searched refs:getNumberOfDimensions (Results 1 – 25 of 42) sorted by relevance

12

/packages/modules/NeuralNetworks/common/operations/
DUnidirectionalSequenceLSTM.cpp201 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
209 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
214 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
220 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
226 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
230 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
236 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
242 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
246 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
260 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()
[all …]
DQLSTM.cpp180 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
187 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
192 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
198 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
204 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
208 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
214 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
220 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
224 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
238 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()
[all …]
DBidirectionalSequenceRNN.cpp393 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
394 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2); in prepare()
395 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2); in prepare()
396 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1); in prepare()
397 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2); in prepare()
398 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2); in prepare()
399 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2); in prepare()
400 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1); in prepare()
401 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2); in prepare()
420 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
[all …]
DL2Normalization.cpp55 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
82 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
114 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8SignedImpl()
141 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32()
170 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8()
186 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8Signed()
229 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
239 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
DStridedSlice.cpp65 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in compute()
135 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
144 uint32_t numInputDims = getNumberOfDimensions(inputShape); in prepare()
151 NN_OPS_CHECK(getNumberOfDimensions(beginShape) == 1); in prepare()
152 NN_OPS_CHECK(getNumberOfDimensions(endShape) == 1); in prepare()
153 NN_OPS_CHECK(getNumberOfDimensions(stridesShape) == 1); in prepare()
DReduce.cpp56 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute()
85 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateProdSum()
108 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateMaxMin()
124 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateLogical()
132 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
138 NN_RET_CHECK_EQ(getNumberOfDimensions(axesShape), 1u); in prepare()
DSqueeze.cpp64 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
78 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in prepare()
80 NN_RET_CHECK_LE(getNumberOfDimensions(inputShape), 4); in prepare()
84 NN_OPS_CHECK(getNumberOfDimensions(squeezeDimsShape) == 1); in prepare()
DSlice.cpp59 std::vector<uint32_t> outputIndex(getNumberOfDimensions(outputShape), 0); in evalGeneric()
61 std::vector<uint32_t> inputIndex(getNumberOfDimensions(inputShape)); in evalGeneric()
112 const int32_t n_dims = getNumberOfDimensions(inputShape); in prepare()
116 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1); in prepare()
120 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()
DUnidirectionalSequenceRNN.cpp169 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
170 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in prepare()
171 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentWeights), 2); in prepare()
172 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
173 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2); in prepare()
DSoftmax.cpp59 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxSlowFloat32()
87 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxFloat32()
134 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxQuant8Impl()
205 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxQuant8()
254 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
277 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepare()
DConcatenation.cpp63 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
88 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
171 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate()
186 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare()
195 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
DSimpleMath.cpp52 int32_t* scratchBuffer = new int32_t[getNumberOfDimensions(inputShape)]; in meanGeneric()
67 getNumberOfDimensions(inputShape), outputData, in meanGeneric()
69 getNumberOfDimensions(outputShape), axis, axisSize, keepDims, scratchBuffer, in meanGeneric()
DLocalResponseNormalization.cpp59 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
89 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNorm()
175 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
185 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
DGather.cpp45 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
90 output.dimensions.reserve(getNumberOfDimensions(input) + getNumberOfDimensions(indices) - 1); in prepare()
DDequantize.cpp55 for (int i = getNumberOfDimensions(inputShape) - 1; i > channelDim; --i) { in computePerChannel()
87 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
108 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepare()
DTranspose.cpp94 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
108 uint32_t numInputDims = getNumberOfDimensions(input); in prepare()
127 NN_RET_CHECK_EQ(getNumberOfDimensions(permShape), 1); in prepare()
DFullyConnected.cpp198 NN_RET_CHECK_GE(getNumberOfDimensions(input), 2); in validateShapes()
199 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateShapes()
200 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in validateShapes()
201 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in validateShapes()
DGenerateProposals.cpp237 NN_RET_CHECK_EQ(getNumberOfDimensions(roiShape), 2); in prepare()
238 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 2); in prepare()
239 NN_RET_CHECK_EQ(getNumberOfDimensions(batchesShape), 1); in prepare()
240 NN_RET_CHECK_EQ(getNumberOfDimensions(imageInfoShape), 2); in prepare()
768 NN_RET_CHECK(getNumberOfDimensions(scoreShape) == 2); in prepare()
769 NN_RET_CHECK(getNumberOfDimensions(roiShape) == 2); in prepare()
770 NN_RET_CHECK(getNumberOfDimensions(batchesShape) == 1); in prepare()
1299 NN_RET_CHECK_EQ(getNumberOfDimensions(scoreShape), 4); in prepare()
1300 NN_RET_CHECK_EQ(getNumberOfDimensions(bboxDeltasShape), 4); in prepare()
1301 NN_RET_CHECK_EQ(getNumberOfDimensions(anchorsShape), 2); in prepare()
[all …]
DChannelShuffle.cpp43 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
71 NN_RET_CHECK_LE(getNumberOfDimensions(inputShape), 4); in validate()
DElementwise.cpp122 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateFloor()
138 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepareFloor()
DConv2D.cpp539 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
540 const auto filterRank = getNumberOfDimensions(context->getInputShape(kFilterTensor)); in validate()
653 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
654 NN_RET_CHECK_EQ(getNumberOfDimensions(filter), 4); in prepare()
655 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
DTransposeConv2D.cpp78 NN_RET_CHECK_EQ(getNumberOfDimensions(context->getInputShape(3)), 1); in initialize()
505 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 4); in prepare()
506 NN_RET_CHECK_EQ(getNumberOfDimensions(filter), 4); in prepare()
507 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in prepare()
/packages/modules/NeuralNetworks/common/
DOperationsUtils.cpp154 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions() function
341 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in calculateBroadcastedShape()
342 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in calculateBroadcastedShape()
421 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
440 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
461 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
462 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
469 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
480 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
481 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
[all …]
/packages/modules/NeuralNetworks/common/include/
DCpuOperationUtils.h120 NN_RET_CHECK_EQ(getNumberOfDimensions(nchwShape), 4) in convertNchwToNhwc()
141 NN_RET_CHECK_EQ(getNumberOfDimensions(nhwcShape), 4) in convertNhwcToNchw()
187 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4); in initialize()
DOperationsUtils.h135 uint32_t getNumberOfDimensions(const Shape& shape);
145 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
327 const int numDims = getNumberOfDimensions(shape); in transposeFirstTwoDimensions()
348 NN_RET_CHECK(getNumberOfDimensions(shape) >= 2); in transposeFirstTwoDimensions()

12