Home
last modified time | relevance | path

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

12

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceLSTM.cpp198 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
206 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
211 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
217 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
223 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
227 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
233 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
239 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
243 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
257 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()
[all …]
DQLSTM.cpp179 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
186 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToOutputShape), 2); in prepare()
191 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToOutputShape), 2); in prepare()
197 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToInputShape), 2); in prepare()
203 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToForgetShape), 2); in prepare()
207 NN_RET_CHECK_EQ(getNumberOfDimensions(inputToCellShape), 2); in prepare()
213 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToInputShape), 2); in prepare()
219 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToForgetShape), 2); in prepare()
223 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentToCellShape), 2); in prepare()
237 NN_RET_CHECK_EQ(getNumberOfDimensions(cellToInputShape), 1); in prepare()
[all …]
DBidirectionalSequenceRNN.cpp394 NN_RET_CHECK_EQ(getNumberOfDimensions(input), 3); in prepare()
395 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2); in prepare()
396 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2); in prepare()
397 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1); in prepare()
398 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2); in prepare()
399 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2); in prepare()
400 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2); in prepare()
401 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1); in prepare()
402 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2); in prepare()
421 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
[all …]
DStridedSlice.cpp64 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in compute()
133 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
141 uint32_t numInputDims = getNumberOfDimensions(inputShape); in prepare()
148 NN_OPS_CHECK(getNumberOfDimensions(beginShape) == 1); in prepare()
149 NN_OPS_CHECK(getNumberOfDimensions(endShape) == 1); in prepare()
150 NN_OPS_CHECK(getNumberOfDimensions(stridesShape) == 1); in prepare()
DSqueeze.cpp67 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
80 int32_t numInputDims = static_cast<int32_t>(getNumberOfDimensions(inputShape)); in prepare()
82 NN_RET_CHECK_LE(getNumberOfDimensions(inputShape), 4); in prepare()
86 NN_OPS_CHECK(getNumberOfDimensions(squeezeDimsShape) == 1); in prepare()
DL2Normalization.cpp54 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normFloat32Impl()
81 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8Impl()
113 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in l2normQuant8SignedImpl()
140 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32()
169 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8()
185 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8Signed()
226 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
234 int32_t numDimensions = getNumberOfDimensions(input); 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()
DSlice.cpp58 std::vector<uint32_t> outputIndex(getNumberOfDimensions(outputShape), 0); in evalGeneric()
60 std::vector<uint32_t> inputIndex(getNumberOfDimensions(inputShape)); in evalGeneric()
107 const int32_t n_dims = getNumberOfDimensions(inputShape); in prepare()
111 NN_RET_CHECK_EQ(getNumberOfDimensions(beginShape), 1); in prepare()
115 NN_RET_CHECK_EQ(getNumberOfDimensions(sizeShape), 1); in prepare()
DReduce.cpp56 const uint32_t inputRank = getNumberOfDimensions(inputShape); in compute()
84 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateProdSum()
107 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateMaxMin()
123 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateLogical()
130 const uint32_t inputRank = getNumberOfDimensions(inputShape); in prepare()
136 NN_RET_CHECK_EQ(getNumberOfDimensions(axesShape), 1u); 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()
DConcatenation.cpp61 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
86 getNumberOfDimensions(outputShape) - axis - 1, inputDataPtrs.data(), in concatenation()
167 const uint32_t inputRank = getNumberOfDimensions(context->getInputShape(i)); in validate()
180 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare()
189 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
DSoftmax.cpp57 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxSlowFloat32()
85 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxFloat32()
132 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in softmaxQuant8Impl()
203 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxQuant8()
250 const auto inputRank = getNumberOfDimensions(context->getInputShape(kInputTensor)); in validate()
271 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepare()
DLocalResponseNormalization.cpp58 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in localResponseNormFloat32Impl()
88 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNorm()
171 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
179 int32_t numDimensions = getNumberOfDimensions(input); in prepare()
DGather.cpp48 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
93 output.dimensions.reserve(getNumberOfDimensions(input) + getNumberOfDimensions(indices) - 1); in prepare()
DTranspose.cpp92 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
104 uint32_t numInputDims = getNumberOfDimensions(input); in prepare()
123 NN_RET_CHECK_EQ(getNumberOfDimensions(permShape), 1); in prepare()
DDequantize.cpp57 for (int i = getNumberOfDimensions(inputShape) - 1; i > channelDim; --i) { in computePerChannel()
88 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validate()
109 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepare()
DFullyConnected.cpp196 NN_RET_CHECK_GE(getNumberOfDimensions(input), 2); in validateShapes()
197 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateShapes()
198 NN_RET_CHECK_EQ(getNumberOfDimensions(weights), 2); in validateShapes()
199 NN_RET_CHECK_EQ(getNumberOfDimensions(bias), 1); in validateShapes()
DChannelShuffle.cpp46 getNumberOfElements(inputShape, axis + 1, getNumberOfDimensions(inputShape)); in eval()
74 NN_RET_CHECK_LE(getNumberOfDimensions(inputShape), 4); in validate()
DExpandDims.cpp28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
DElementwise.cpp126 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in validateFloor()
144 NN_RET_CHECK_LE(getNumberOfDimensions(input), 4); in prepareFloor()
DRank.cpp60 getNumberOfDimensions(context->getInputShape(kInputTensor)); in execute()
/frameworks/ml/nn/common/
DOperationsUtils.cpp156 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions() function
343 uint32_t numberOfDims1 = getNumberOfDimensions(in1); in calculateBroadcastedShape()
344 uint32_t numberOfDims2 = getNumberOfDimensions(in2); in calculateBroadcastedShape()
423 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in depthToSpacePrepare()
442 NN_OPS_CHECK(getNumberOfDimensions(input) == 4); in spaceToDepthPrepare()
463 NN_OPS_CHECK(getNumberOfDimensions(valueShape) >= 2); in embeddingLookupPrepare()
464 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in embeddingLookupPrepare()
471 for (uint32_t i = 2; i < getNumberOfDimensions(valueShape); i++) { in embeddingLookupPrepare()
482 NN_OPS_CHECK(getNumberOfDimensions(lookupShape) == 1); in hashtableLookupPrepare()
483 NN_OPS_CHECK(getNumberOfDimensions(keyShape) == 1); in hashtableLookupPrepare()
[all …]
/frameworks/ml/nn/common/include/
DCpuOperationUtils.h119 NN_RET_CHECK_EQ(getNumberOfDimensions(nchwShape), 4) in convertNchwToNhwc()
140 NN_RET_CHECK_EQ(getNumberOfDimensions(nhwcShape), 4) in convertNhwcToNchw()
186 NN_RET_CHECK_EQ(getNumberOfDimensions(shape), 4); in initialize()
DOperationsUtils.h151 uint32_t getNumberOfDimensions(const Shape& shape);
161 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
343 const int numDims = getNumberOfDimensions(shape); in transposeFirstTwoDimensions()
364 NN_RET_CHECK(getNumberOfDimensions(shape) >= 2); in transposeFirstTwoDimensions()
/frameworks/base/media/mca/filterfw/java/android/filterfw/core/
DNativeFrame.java170 if (getFormat().getNumberOfDimensions() != 2) { in setBitmap()
187 if (getFormat().getNumberOfDimensions() != 2) { in getBitmap()

12