Home
last modified time | relevance | path

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

12

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceLSTM.cpp201 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
209 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
226 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
230 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
246 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
298 NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1); in prepare()
301 NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1); in prepare()
304 NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1); in prepare()
309 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2); in prepare()
321 NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2); in prepare()
[all …]
H A DQLSTM.cpp180 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
187 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
204 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
208 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
224 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
276 NN_RET_CHECK_EQ(getNumberOfDimensions(forgetGateBiasShape), 1); in prepare()
279 NN_RET_CHECK_EQ(getNumberOfDimensions(cellGateBiasShape), 1); in prepare()
282 NN_RET_CHECK_EQ(getNumberOfDimensions(outputGateBiasShape), 1); in prepare()
287 NN_RET_CHECK_EQ(getNumberOfDimensions(projectionShape), 2); in prepare()
299 NN_RET_CHECK_EQ(getNumberOfDimensions(outputStateShape), 2); in prepare()
[all …]
H A DBidirectionalSequenceRNN.cpp393 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
394 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 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()
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()
421 NN_RET_CHECK_EQ(getNumberOfDimensions(fwAuxWeights), 2); in prepare()
422 NN_RET_CHECK_EQ(getNumberOfDimensions(bwAuxWeights), 2); in prepare()
[all …]
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A 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()
H A DConcatenation.cpp63 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
88 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(),
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()
H A 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()
H A DLocalResponseNormalization.cpp59 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
89 int32_t ndim = getNumberOfDimensions(inputShape);
175 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
185 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
H A 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()
H A DGather.cpp45 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
90 output.dimensions.reserve(getNumberOfDimensions(input) + getNumberOfDimensions(indices) - 1); in prepare()
H A 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()
H A 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()
H A DChannelShuffle.cpp43 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
71 NN_RET_CHECK_LE(getNumberOfDimensions(inputShape), 4); in validate()
H A DExpandDims.cpp28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
H A DElementwise.cpp122 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateFloor()
138 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepareFloor()
H A DRank.cpp59 getNumberOfDimensions(context->getInputShape(kInputTensor)); in execute()
/aosp12/packages/modules/NeuralNetworks/common/
H A DOperationsUtils.cpp154 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions() function
341 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in calculateBroadcastedShape()
421 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
440 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
481 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
530 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in batchToSpacePrepare()
557 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToBatchPrepare()
676 if (getNumberOfDimensions(input) > 1) { in argMinMaxPrepare()
723 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in groupedConvPrepare()
724 NN_OPS_CHECK(getNumberOfDimensions(filter) == 4); in groupedConvPrepare()
[all …]
/aosp12/packages/modules/NeuralNetworks/common/include/
H A 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()
H A 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()
/aosp12/frameworks/base/media/mca/filterfw/java/android/filterfw/core/
H A DNativeFrame.java170 if (getFormat().getNumberOfDimensions() != 2) { in setBitmap()
187 if (getFormat().getNumberOfDimensions() != 2) { in getBitmap()

12